fixing messages

main.py

@@ -135,13 +135,7 @@ def predict(data: PredictRequest):
 
     messages = [
       {"role": "system", "content": "You are a helpful assistant with vision abilities."},
-      {
-        "role": "user",
-        "content": [
-            {"type": "image", "image": f"data:image;base64,{image}"} for image in data.image_base64
-        ]
-        + [{"type": "text", "text": data.prompt}],
-     },
+      {"role": "user", "content": [{"type": "image", "image": data.image_base64}, {"type": "text", "text": data.prompt}]},
     ]
     
     print("messages: ", messages)

model.py

@@ -28,10 +28,12 @@ predict_payload = {
     "prompt": "describe the image",
 }
 
+print("Payload:", predict_payload)
+
 predict_response = requests.post(f"{BASE_URL}/predict", json=predict_payload)
 
 
-# Step 4: Print the response
+# # Step 4: Print the response
 if predict_response.status_code == 200:
     print("Response:", predict_response.json())
 else: