manpk-ai
/

OpenCUA-7B

Image-Text-to-Text

feature-extraction

Computer-Use-Agent

Model card Files Files and versions

manpk-ai commited on Sep 19

Commit

2f8e98a

·

1 Parent(s): 4214e4e

bug fix

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -105,8 +105,8 @@ class EndpointHandler():
                 for message in messages:
                     if isinstance(message.get("content"), list):
                         for content_item in message["content"]:
-                            if content_item.get("type") == "image":
-                                image_data = content_item.get("image", "")
                                 if image_data.startswith("data:image"):
                                     # Decode base64 image
                                     image_data = image_data.split(",")[1]

                 for message in messages:
                     if isinstance(message.get("content"), list):
                         for content_item in message["content"]:
+                            if content_item.get("type") == "image_url":
+                                image_data = content_item.get("image_url").get("url", "")
                                 if image_data.startswith("data:image"):
                                     # Decode base64 image
                                     image_data = image_data.split(",")[1]