Jeney
/

vilt-b32-finetuned-vqa

Visual Question Answering

Inference Endpoints

Model card Files Files and versions Community

Jeney commited on Jul 1, 2023

Commit

7807370

•

1 Parent(s): 1de5e91

Add outputs

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -26,4 +26,4 @@ class EndpointHandler:
         # postprocess the prediction
         logits = outputs.logits
         idx = logits.argmax(-1).item()
-        return [{"answer": self.model.config.id2label[idx]}]

         # postprocess the prediction
         logits = outputs.logits
         idx = logits.argmax(-1).item()
+        return [{"best_answer": self.model.config.id2label[idx], "outputs":outputs}]