Update handler.py

handler.py

@@ -59,15 +59,17 @@ class EndpointHandler():
         #     })
         inputs = data.pop("inputs", data)
         parameters = data.pop("parameters", {})
-
-        try:
-            imageBase64 = inputs["image"]
-            # image = Image.open(BytesIO(base64.b64decode(imageBase64.split(",")[1].encode())))
-            image = Image.open(BytesIO(base64.b64decode(imageBase64)))
-
-        except:
-            image_url = inputs['image']
-            image = Image.open(requests.get(image_url, stream=True).raw).convert('RGB')
+        if isinstance(inputs, Image.Image):
+            image = [inputs]
+        else:
+            try:
+                imageBase64 = inputs["image"]
+                # image = Image.open(BytesIO(base64.b64decode(imageBase64.split(",")[1].encode())))
+                image = Image.open(BytesIO(base64.b64decode(imageBase64)))
+    
+            except:
+                image_url = inputs['image']
+                image = Image.open(requests.get(image_url, stream=True).raw).convert('RGB')
             
         question = inputs["question"]