Spaces:
Running
Running
Commit
·
52c90eb
1
Parent(s):
dbcff35
adding change
Browse files
main.py
CHANGED
@@ -134,15 +134,17 @@ def predict(data: PredictRequest):
|
|
134 |
# ]
|
135 |
|
136 |
messages = [
|
137 |
-
|
138 |
-
|
139 |
-
|
140 |
-
|
141 |
-
|
142 |
-
|
143 |
-
|
144 |
-
|
145 |
-
]
|
|
|
|
|
146 |
|
147 |
|
148 |
# Prepare inputs for the model
|
|
|
134 |
# ]
|
135 |
|
136 |
messages = [
|
137 |
+
{"role": "system", "content": "You are a helpful assistant with vision abilities."},
|
138 |
+
{
|
139 |
+
"role": "user",
|
140 |
+
"content": [
|
141 |
+
{"type": "image", "image": image} for image in data.image_base64
|
142 |
+
]
|
143 |
+
+ [{"type": "text", "text": data.prompt}],
|
144 |
+
},
|
145 |
+
]
|
146 |
+
|
147 |
+
print("messages: ", messages)
|
148 |
|
149 |
|
150 |
# Prepare inputs for the model
|