Spaces:
Running
Running
Commit
·
c229531
1
Parent(s):
52c90eb
fixing prompt
Browse files
main.py
CHANGED
@@ -136,12 +136,12 @@ def predict(data: PredictRequest):
|
|
136 |
messages = [
|
137 |
{"role": "system", "content": "You are a helpful assistant with vision abilities."},
|
138 |
{
|
139 |
-
|
140 |
-
|
141 |
-
|
142 |
-
|
143 |
-
|
144 |
-
|
145 |
]
|
146 |
|
147 |
print("messages: ", messages)
|
|
|
136 |
messages = [
|
137 |
{"role": "system", "content": "You are a helpful assistant with vision abilities."},
|
138 |
{
|
139 |
+
"role": "user",
|
140 |
+
"content": [
|
141 |
+
{"type": "image", "image": f"data:image;base64,{image}"} for image in data.image_base64
|
142 |
+
]
|
143 |
+
+ [{"type": "text", "text": data.prompt}],
|
144 |
+
},
|
145 |
]
|
146 |
|
147 |
print("messages: ", messages)
|