messages

main.py

@@ -123,20 +123,16 @@ def predict(data: PredictRequest):
 
 
     # Create the input message structure
-    # messages = [
-    #     {
-    #         "role": "user",
-    #         "content": [
-    #             {"type": "image", "image": f"data:image;base64,{data.image_base64}"},
-    #             {"type": "text", "text": data.prompt},
-    #         ],
-    #     }
-    # ]
-
     messages = [
-      {"role": "system", "content": "You are a helpful assistant with vision abilities."},
-      {"role": "user", "content": [{"type": "image", "image": data.image_base64}, {"type": "text", "text": data.prompt}]},
+        {
+            "role": "user",
+            "content": [
+                {"type": "image", "image": f"data:image;base64,{data.image_base64}"},
+                {"type": "text", "text": data.prompt},
+            ],
+        }
     ]
+
     
     print("messages: ", messages)