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

fixing prompt

Browse files
Files changed (1) hide show
  1. main.py +6 -6
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
- "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)
 
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)