Update handler.py

handler.py

@@ -45,7 +45,15 @@ class EndpointHandler():
 
     def __call__(self, data: Any) -> List[List[Dict[str, float]]]:
         image = data['inputs'].pop("image", None)
+        image_url = data['inputs'].pop("image_url", None)
         type = data['inputs'].pop("type", '<MORE_DETAILED_CAPTION>')
         text = data['inputs'].pop("text", None)
-        image = Image.open(BytesIO(base64.b64decode(image)))
+        if image:
+            image = Image.open(BytesIO(base64.b64decode(image)))
+        elif image_url:
+            response = requests.get(image_url)
+            if response.status_code == 200:
+                image = Image.open(BytesIO(response.content))
+            else:
+                raise ValueError(f"Unable to download image from URL: {image_url}")
         return self.run_example(image, type, text_input=text)