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,
|