Upload handler.py

handler.py

@@ -28,12 +28,12 @@ class EndpointHandler():
         # }
         
         inputs = data.get("inputs")
-        imageURL = inputs.get("image")
+        imageBase64 = inputs.get("image")
         text = inputs.get("text")
-        print(imageURL)
-        print(text)
-        image = Image.open(requests.get(imageURL, stream=True).raw)
-        # image = Image.open(BytesIO(base64.b64decode(imageData)))
+        # print(imageURL)
+        # print(text)
+        # image = Image.open(requests.get(imageURL, 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)
         embeddings = outputs.image_embeds.detach().numpy().flatten().tolist()