Update imageBase64

handler.py

@@ -18,12 +18,21 @@ class EndpointHandler():
         # inputs = self.processor(text=["a photo of a cat", "a photo of a dog"], images=image, return_tensors="pt", padding=True)
         # logits_per_image = outputs.logits_per_image # this is the image-text similarity score
         # probs = logits_per_image.softmax(dim=1) # we can take the softmax to get the label probabilities
-
+        # print(probs)
+        
+        # data = {
+        #     "inputs": {
+        #         "image": "http://images.cocodataset.org/val2017/000000039769.jpg",
+        #         "text": ["a photo of a cat", "a photo of a dog"]
+        #     },
+        # }
+        
         inputs = data.get("inputs")
+        imageURL = inputs.get("image")
         text = inputs.get("text")
-        imageData = inputs.get("image")
-        url = inputs.get("image")
-        image = Image.open(requests.get(url, stream=True).raw)
+        print(imageURL)
+        print(text)
+        image = Image.open(requests.get(imageURL, stream=True).raw)
         # image = Image.open(BytesIO(base64.b64decode(imageData)))
         inputs = self.processor(text=text, images=image, return_tensors="pt", padding=True)
         outputs = self.model(**inputs)