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