danilohssantana commited on
Commit
52c90eb
·
1 Parent(s): dbcff35

adding change

Browse files
Files changed (1) hide show
  1. main.py +11 -9
main.py CHANGED
@@ -134,15 +134,17 @@ def predict(data: PredictRequest):
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
 
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