Spaces:
Runtime error
Runtime error
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
|