Update handler.py

handler.py

@@ -23,8 +23,17 @@ class EndpointHandler():
         imageData = inputs.get("image")
         print("** imageData: ", imageData)
 
-        image = Image.open(BytesIO(base64.b64decode(imageData)))
-        print("** image: ", image)
+        image = None
+        if image_data:
+            try:
+                image = Image.open(BytesIO(base64.b64decode(image_data)))
+                print("** image: ", image)
+            except Exception as e:
+                raise ValueError(f"Error decoding image: {e}")
+
+        if not text and not image:
+            raise ValueError("Both text and image cannot be None. Provide at least one.")
+
 
         inputs = self.processor(text=text, images=image, return_tensors="pt", padding=True)
         outputs = self.model(**inputs)