moooji
/

blip-vqa-capfilt-large

Inference Endpoints

Model card Files Files and versions Community

moooji commited on Feb 8, 2023

Commit

448efba

•

1 Parent(s): acbc291

Update handler.py

Files changed (1) hide show

handler.py +3 -3

handler.py CHANGED Viewed

@@ -16,8 +16,8 @@ class EndpointHandler():
         inputs = data.pop("inputs", data)
         image = Image.open(BytesIO(base64.b64decode(inputs['image'])))
-        inputs = self.processor(image, return_tensors="pt").to(device)
-        outputs = self.model(**inputs)
-        return outputs.toList()

         inputs = data.pop("inputs", data)
         image = Image.open(BytesIO(base64.b64decode(inputs['image'])))
+        inputs = self.processor(image, inputs['question'], return_tensors="pt").to(device)
+        outputs = self.model.generate(**inputs)
+        return processor.decode(out[0], skip_special_tokens=True)