Update app.py

app.py

@@ -14,7 +14,7 @@ def analyze_sentence(index):
     attn_map_shape = row['attention_maps_shape'][1:]
     seq_len = attn_map_shape[1]    
     attn_maps = np.array(row['attention_maps']).reshape(*attn_map_shape)
-    plot = sns.heatmap(attn_maps.sum(0))
+    plot = sns.heatmap(attn_maps.sum(0)[1:, 1:])
     plt.xticks(np.arange(seq_len - 1) + 0.5,
            tokenizer.tokenize(text, add_special_tokens=False), rotation=90);
     plt.yticks(np.arange(seq_len - 1) + 0.5,
@@ -25,6 +25,6 @@ def analyze_sentence(index):
     return text, plot
 
 
-iface = gr.Interface(fn=analyze_sentence, inputs=[gr.Dropdown(choices=dataset['text'], type='index')], 
+iface = gr.Interface(fn=analyze_sentence, inputs=[gr.Dropdown(choices=dataset['text'], index=0, type='index')], 
                      outputs=[gr.Label(), gr.Plot(label="Plot")])
 iface.launch()