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

5edf3dc

•

1 Parent(s): d576d35

Updated handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -31,7 +31,7 @@ class EndpointHandler:
         device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
         if type(inputs) is str:
-            inputs = inputs.encode('utf-8')
         raw_images = Image.open(BytesIO(inputs))

         device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
         if type(inputs) is str:
+            inputs = base64.b64encode(bytes(inputs, 'utf-8'))
         raw_images = Image.open(BytesIO(inputs))