paragon-AI
/

blip2-image-to-text

visual-question-answering

image-captioning

Inference Endpoints

Model card Files Files and versions Community

thoth-AI commited on Jun 24, 2023

Commit

70c6e5e

•

1 Parent(s): d6460d1

Updated handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -32,7 +32,7 @@ class EndpointHandler:
         # parameters = data.pop("parameters", {})
         device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-        raw_images = Image.open(BytesIO(inputs))
         processed_image = self.processor(images=raw_images, return_tensors="pt").to(device)

         # parameters = data.pop("parameters", {})
         device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+        raw_images = Image.open(BytesIO(inputs.encode('utf-8')))
         processed_image = self.processor(images=raw_images, return_tensors="pt").to(device)