added attention vizualization and qa model

app.py

@@ -13,9 +13,9 @@ nltk.download('popular')
 #   Constants
 #############################
 
-examples = [["What's the meaning of life?", "eli5", "constraint"],
-            ["boat, water, bird", "commongen", "constraint"],
-            ["What flows under a bridge?", "commonsense_qa", "constraint"]]
+#examples = [["What's the meaning of life?", "eli5", "constraint"],
+#            ["boat, water, bird", "commongen", "constraint"],
+#            ["What flows under a bridge?", "commonsense_qa", "constraint"]]
 
 commongen_bart = RelationsInference(
     model_path='MrVicente/commonsense_bart_commongen',
@@ -108,7 +108,7 @@ with app:
         vis_btn = gr.Button(value="See Attention Scores")
     model_btn.click(fn=infer_bart, inputs=[context_input, task_type_choice, decoding_type_choice],
                     outputs=[model_result_output])
-    vis_btn.click(fn=plot_attention, inputs=[layer, head], outputs=[plot_output])
+    vis_btn.click(fn=plot_attention, inputs=[context_input, task_type_choice, layer, head], outputs=[plot_output])
 
 if __name__ == '__main__':
     app.launch()