Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ def analyze_sentence(index):
|
|
17 |
attn_map_shape = row['attention_maps_shape'][1:]
|
18 |
seq_len = attn_map_shape[1]
|
19 |
attn_maps = np.array(row['attention_maps']).reshape(*attn_map_shape).clip(0, 1)
|
20 |
-
fig = plt.figure(figsize=(
|
21 |
sns.heatmap(attn_maps.sum(0)[1:, 1:])
|
22 |
plt.xticks(np.arange(seq_len - 1) + 0.5, tokenized[1:], rotation=90);
|
23 |
plt.yticks(np.arange(seq_len - 1) + 0.5, tokenized[1:], rotation=0);
|
@@ -31,7 +31,7 @@ with demo:
|
|
31 |
with gr.Row():
|
32 |
dropdown = gr.Dropdown(choices=dataset['text'], value=0, min_width=750, type='index')
|
33 |
btn = gr.Button("Run")
|
34 |
-
output = gr.Plot(label="Plot", container=
|
35 |
btn.click(analyze_sentence, [dropdown], [output])
|
36 |
|
37 |
|
|
|
17 |
attn_map_shape = row['attention_maps_shape'][1:]
|
18 |
seq_len = attn_map_shape[1]
|
19 |
attn_maps = np.array(row['attention_maps']).reshape(*attn_map_shape).clip(0, 1)
|
20 |
+
fig = plt.figure(figsize=(0.2 + 0.55 * len(tokenized), 0.5 * len(tokenized)))
|
21 |
sns.heatmap(attn_maps.sum(0)[1:, 1:])
|
22 |
plt.xticks(np.arange(seq_len - 1) + 0.5, tokenized[1:], rotation=90);
|
23 |
plt.yticks(np.arange(seq_len - 1) + 0.5, tokenized[1:], rotation=0);
|
|
|
31 |
with gr.Row():
|
32 |
dropdown = gr.Dropdown(choices=dataset['text'], value=0, min_width=750, type='index')
|
33 |
btn = gr.Button("Run")
|
34 |
+
output = gr.Plot(label="Plot", container=True)
|
35 |
btn.click(analyze_sentence, [dropdown], [output])
|
36 |
|
37 |
|