Spaces:

SeyedAli
/

Persian-Visual-Question-Answering-1

Runtime error

SeyedAli commited on Sep 25, 2023

Commit

1148ad6

1 Parent(s): b242b72

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -43,7 +43,7 @@ def VQA(image,text):
         outputs = VQA_model(**encoding)
         logits = outputs.logits
         idx = logits.argmax(-1).item()
-        return run_en_fa_transaltion_model(VQA_model.config.id2label[idx])
 iface = gr.Interface(fn=VQA, inputs=["image","text"], outputs="text")
 iface.launch(share=False)

         outputs = VQA_model(**encoding)
         logits = outputs.logits
         idx = logits.argmax(-1).item()
+        return run_en_fa_transaltion_model(VQA_model.config.id2label[idx])[0]
 iface = gr.Interface(fn=VQA, inputs=["image","text"], outputs="text")
 iface.launch(share=False)