Update handler.py

handler.py

@@ -36,9 +36,9 @@ class EndpointHandler():
         # image = Image.open(requests.get(imageBase64, stream=True).raw)
         base64_data = imageBase64.split(",")[1]
         image_data = base64.b64decode(base64_data)
-        
+        # print(image_data)
         # Open the image
-        image = Image.open(BytesIO(image_data))
+        image = Image.open((image_data))
         # 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)