Upload handler.py

handler.py

@@ -30,6 +30,17 @@ class EndpointHandler():
         text = data.get("text", "")
         parameters = data.pop("parameters", {})
 
+        try:
+            # Ensure inputs is a list of image bytes, even if only a single image is provided
+            if isinstance(images, bytes):  # Single image as bytes
+                raw_images = [Image.open(BytesIO(images))]
+            elif isinstance(images, list):  # Multiple images as list of bytes
+                raw_images = [Image.open(BytesIO(_img)) for _img in images if isinstance(_img, bytes)]
+            else:
+                raise ValueError("Invalid image input format. Expected bytes or list of bytes.")
+        except Exception as e:
+            return {"error": f"Error fetching or processing image: {str(e)}"}
+
         processed_image = self.processor(images=images, text=text, return_tensors="pt")
         processed_image["pixel_values"] = processed_image["pixel_values"].to(device)
         processed_image = {**processed_image, **parameters}
@@ -43,4 +54,3 @@ class EndpointHandler():
         return {"description": description}
 
 
-handler = EndpointHandler()