fixed viz function parameter error

app.py

@@ -40,7 +40,7 @@ def infer_bart(context, task_type, decoding_type_str):
         if decoding_type_str =='default':
             response, _, _ = commongen_bart.generate_based_on_context(context, use_kg=False)
         else:
-            response, _, _ = commongen_bart.generate_contrained_based_on_context([context], use_kg=True)
+            response, _, _ = commongen_bart.generate_contrained_based_on_context([context], use_kg=True, max_concepts=2)
     elif Data_Type(task_type) == Data_Type.ELI5:
         response, _, _ = qa_bart.generate_based_on_context(context, use_kg=False)
     else:

attention_viz.py

@@ -170,7 +170,7 @@ class AttentionVisualizer:
         plt.title(title)
         plt.show()
 
-    def plot_attn_lines_concepts_ids(title, examples, layer, head,
+    def plot_attn_lines_concepts_ids(self, title, examples, layer, head,
                                      relations_total, width=3, example_sep=3,
                                      word_height=1, pad=0.1, hide_sep=False):
         # examples -> {'words': tokens, 'attentions': [layer][head]}