fixed send.py

The-Swarm-Corporation · Mar 10, 2024 · ddddb63 · ddddb63
1 parent b0a93f8
commit ddddb63
Showing 1 changed file with 12 additions and 12 deletions.
diff --git a/scripts/send.py b/scripts/send.py
@@ -13,24 +13,24 @@ def image_to_base64(image_path):
 
 # Replace 'image.jpg' with the path to your image
 base64_image = image_to_base64('images/3897e80dcb0601c0.jpg')
+text_data = {
+    "type": "text",
+    "text": "Describe what is in the image"
+}
+image_data = {
+    "type": "image_url",
+    "image_url": {
+        "url": f"data:image/jpeg;base64,{base64_image}"
+    }
+}
+
 # Construct the request data
 request_data = {
     "model": "cogvlm-chat-17b",
     "messages": [
         {
         "role": "user",
-        "content": [
-            {
-            "type": "text",
-            "text": "Whats in this image?"
-            },
-            {
-            "type": "image_url",
-            "image_url": {
-                "url": f"data:image/jpeg;base64,{base64_image}"
-            }
-            }
-        ]
+        "content": [text_data, image_data]
         }
     ],
     "temperature": 0.8,