florentgbelidji
/

blip_captioning

image-captioning

endpoints-template

Inference Endpoints

Model card Files Files and versions Community

florentgbelidji HF staff commited on Oct 19, 2022

Commit

db4cf02

•

1 Parent(s): 9d9b5e2

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -46,7 +46,7 @@ class EndpointHandler():
         parameters = data.pop("parameters", {})
-        image = Image.open(BytesIO(inputs['image']))
         image = self.transform(image).unsqueeze(0).to(device)
         with torch.no_grad():
             caption = self.model.generate(

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