Update handler.py

handler.py

@@ -28,12 +28,12 @@ class EndpointHandler():
         # }
         
         inputs = data.get("inputs")
-        # imageBase64 = inputs.get("image")
-        imageURL = inputs.get("image")
+        imageBase64 = inputs.get("image")
+        # imageURL = inputs.get("image")
         text = inputs.get("text")
         # print(imageURL)
         # print(text)
-        image = Image.open(requests.get(imageURL, stream=True).raw)
+        image = Image.open(requests.get(imageBase64, stream=True).raw)
         # image = Image.open(BytesIO(base64.b64decode(imageBase64.split(",")[1].encode())))
         inputs = self.processor(text=text, images=image, return_tensors="pt", padding=True)
         outputs = self.model(**inputs)