florentgbelidji
/

blip_captioning

image-captioning

endpoints-template

Model card Files Files and versions

florentgbelidji HF Staff commited on Aug 22, 2022

Commit

2ac6664

·

1 Parent(s): a338bba

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -46,7 +46,7 @@ class PreTrainedPipeline():
         parameters = data.pop("parameters", {})
         # decode base64 image to PIL
-        image = Image.open(BytesIO(base64.b64decode(image))
         image = self.transform(image).unsqueeze(0).to(device)
         with torch.no_grad():
             caption = self.model.generate(

         parameters = data.pop("parameters", {})
         # decode base64 image to PIL
+        image = Image.open(BytesIO(base64.b64decode(image)))
         image = self.transform(image).unsqueeze(0).to(device)
         with torch.no_grad():
             caption = self.model.generate(