Update app.py

app.py

@@ -17,7 +17,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).clip(0, 1)
-    fig = plt.figure(figsize=(7, 6))
+    fig = plt.figure(figsize=(6.5, 6))
     sns.heatmap(attn_maps.sum(0)[1:, 1:])
     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);
@@ -29,9 +29,9 @@ def analyze_sentence(index):
 demo = gr.Blocks()
 with demo:
     with gr.Row():
-        dropdown = gr.Dropdown(choices=dataset['text'], value=0, min_width=700, type='index')
+        dropdown = gr.Dropdown(choices=dataset['text'], value=0, min_width=750, type='index')
         btn = gr.Button("Run")
-    output = gr.Plot(label="Plot")
+    output = gr.Plot(label="Plot", container=False)
     btn.click(analyze_sentence, [dropdown], [output])