Update handler.py

handler.py

@@ -59,17 +59,17 @@ class EndpointHandler():
         #     })
         inputs = data.pop("inputs", data)
         parameters = data.pop("parameters", {})
-        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)))
+        # 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')
+            # except:
+        image_url = inputs['image']
+        image = Image.open(requests.get(image_url, stream=True).raw).convert('RGB')
             
         question = inputs["question"]
         
@@ -109,4 +109,4 @@ class EndpointHandler():
         result = {}
         text_output = self.processor.decode(out[0], skip_special_tokens=True)
         result["text_output"] = text_output
-        return result   
+        return text_output