rschwabco
/

clip-vit-base-patch32-embeddings

Model card Files Files and versions

squanchd commited on Jan 31, 2023

Commit

f4e54f2

·

1 Parent(s): ca6c577

pop image

Files changed (1) hide show

handler.py +2 -1

handler.py CHANGED Viewed

@@ -22,7 +22,8 @@ class EndpointHandler():
         """
         words = data.pop("words", data)
-        image = Image.open(BytesIO(base64.b64decode(data['image'])))
         inputs = self.processor(text=words, images=image, return_tensors="pt", padding=True)
         outputs = self.model(**inputs)
         embeddings = outputs.image_embeds.detach().numpy().flatten().tolist()

         """
         words = data.pop("words", data)
+        imageData = data.pop("image", data)
+        image = Image.open(BytesIO(base64.b64decode(imageData)))
         inputs = self.processor(text=words, images=image, return_tensors="pt", padding=True)
         outputs = self.model(**inputs)
         embeddings = outputs.image_embeds.detach().numpy().flatten().tolist()