fix: fixing issues with attention visualization

main.py

@@ -200,8 +200,8 @@ with gr.Blocks(
             xai_interactive = iFrame(
                 label="Interactive Explanation",
                 value=(
-                    '<div style="text-align: center; font-family:arial;"><h4>No Graphic to Display'
-                    " (Yet)</h4></div>"
+                    '<div style="text-align: center; font-family:arial;"><h4>No Graphic'
+                    " to Display (Yet)</h4></div>"
                 ),
                 height="1500px",
                 show_label=True,

utils/formatting.py

@@ -75,5 +75,5 @@ def flatten_attention(values: ndarray, axis: int = 0):
 
 
 def avg_attention(attention_values):
-    attention = attention_values.output_attentions[0][0].detach().numpy()
+    attention = attention_values.decoder_attentions[0][0].detach().numpy()
     return np.mean(attention, axis=0)