trimble
/

clip-vit-large-patch14

Inference Endpoints

Model card Files Files and versions Community

nicklorch commited on Oct 31, 2023

Commit

ff5a99d

•

1 Parent(s): ed041c3

image payload fix

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -33,7 +33,7 @@ class EndpointHandler():
             if image_data is not None:
                 logger.info('image is encoded')
-                image = BytesIO(base64.b64decode(image_data))
             if text_input:
                 processor = self.processor(text=text_input, return_tensors="pt", padding=True).to(device)

             if image_data is not None:
                 logger.info('image is encoded')
+                image = Image.open(BytesIO(base64.b64decode(image_data)))
             if text_input:
                 processor = self.processor(text=text_input, return_tensors="pt", padding=True).to(device)