moooji
/

blip-vqa-capfilt-large

Inference Endpoints

Model card Files Files and versions Community

moooji commited on Feb 8, 2023

Commit

e190089

•

1 Parent(s): cea90f5

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -19,5 +19,5 @@ class EndpointHandler():
         inputs = self.processor(image, inputs['question'], return_tensors="pt").to(device)
         outputs = self.model.generate(**inputs)
-        return self.processor.decode(out[0], skip_special_tokens=True)

         inputs = self.processor(image, inputs['question'], return_tensors="pt").to(device)
         outputs = self.model.generate(**inputs)
+        return self.processor.decode(outputs[0], skip_special_tokens=True)