Spaces:

nielsr
/

vilt-vqa

Running

nielsr HF staff commited on Dec 10, 2021

Commit

12b00d2

•

1 Parent(s): 37ce7af

Fix bug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ def answer_question(image, text):
     logits = outputs.logits
     idx = logits.argmax(-1).item()
-    predicted_answer = model.config.id2label[idx])
     return predicted_answer

     logits = outputs.logits
     idx = logits.argmax(-1).item()
+    predicted_answer = model.config.id2label[idx]
     return predicted_answer