Update handler.py

handler.py

@@ -30,8 +30,12 @@ class EndpointHandler():
         """
         inputs = data.pop("inputs", data)
         parameters = data.pop("parameters", {})
- 
-        raw_images = [Image.open(BytesIO(base64.b64decode(_img))) for _img in inputs]
+
+        if isinstance(inputs, Image.Image):
+            raw_images = [inputs]
+        else:
+            inputs = isinstance(inputs, str) and [inputs] or inputs
+            raw_images = [Image.open(BytesIO(base64.b64decode(_img))) for _img in inputs]
                                      
         processed_images = self.processor(images=raw_images, return_tensors="pt")
         processed_images["pixel_values"] = processed_images["pixel_values"].to(device)