Update app.py

app.py

@@ -25,6 +25,9 @@ def analyze_sentence(index, vis_type):
     attn_maps = attn_maps[:, 1:, 1:]
     if vis_type == VisType.ALL:
         plot_data = attn_maps.sum(0)
+    else:
+        print(vist_type)
+        0/0
     sns.heatmap(plot_data)
     plt.xticks(np.arange(seq_len - 1) + 0.5, tokenized[1:], rotation=90);
     plt.yticks(np.arange(seq_len - 1) + 0.5, tokenized[1:], rotation=0);
@@ -41,7 +44,7 @@ with demo:
                                         choices=[x.split('</s> ')[1] for x in dataset['text']], 
                                value=0, min_width=500, type='index')
         vis_dropdown = gr.Dropdown(label="Visualization", choices=list(VisType), 
-                                   min_width=100, value=VisType.ALL, type='value')
+                                   min_width=200, value=VisType.ALL, type='value')
         btn = gr.Button("Run", min_width=50)
     output = gr.Plot(label="Plot", container=True)
     metrics = gr.Label("Metrics")