Update handler.py

handler.py

@@ -34,12 +34,8 @@ class EndpointHandler():
         # print(imageURL)
         # print(text)
         # 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((image_data))
-        # image = Image.open(BytesIO(base64.b64decode(imageBase64.split(",")[1].encode())))
+        
+        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)
         embeddings = outputs.image_embeds.detach().numpy().flatten().tolist()