Update app.py

app.py

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