Update README.md
Browse files
README.md
CHANGED
|
@@ -45,6 +45,18 @@ processor = PaliGemmaProcessor.from_pretrained(model_id)
|
|
| 45 |
|
| 46 |
image = Image.open("path to your image").convert("RGB")
|
| 47 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 48 |
|
| 49 |
inputs = processor(
|
| 50 |
text=prompt,
|
|
|
|
| 45 |
|
| 46 |
image = Image.open("path to your image").convert("RGB")
|
| 47 |
|
| 48 |
+
prompt = (
|
| 49 |
+
f"Generate a caption for the following coffee maker image. The caption has to be of the following structure:\n"
|
| 50 |
+
"\"A <color> <type>, <accessories>, <shape> shaped, with <screen> and <number> <b_color> butons\"\n\n"
|
| 51 |
+
"in which:\n"
|
| 52 |
+
"- color: red, black, blue...\n"
|
| 53 |
+
"- type: coffee machine, coffee maker, espresso coffee machine...\n"
|
| 54 |
+
"- accessories: a list of accessories like the ones described above\n"
|
| 55 |
+
"- shape: cubed, round...\n"
|
| 56 |
+
"- screen: screen, no screen.\n"
|
| 57 |
+
"- number: amount of buttons to add\n"
|
| 58 |
+
"- b_color: color of the buttons"
|
| 59 |
+
)
|
| 60 |
|
| 61 |
inputs = processor(
|
| 62 |
text=prompt,
|