Jeney
/

vilt-b32-finetuned-vqa

Visual Question Answering

Inference Endpoints

Model card Files Files and versions Community

Jeney commited on Jul 1, 2023

Commit

db02aa7

•

1 Parent(s): 7807370

Output logits

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 [{"best_answer": self.model.config.id2label[idx], "outputs":outputs}]

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