dar-tau commited on
Commit
9fa205f
1 Parent(s): 8b3b771

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ def analyze_sentence(index, vis_type):
35
  plt.xlabel('SOURCE')
36
  plt.grid()
37
  metrics = {k: v for k, v in row.items() if k not in ['text', 'attention_maps', 'attention_maps_shape']}
38
- return fig
39
 
40
  demo = gr.Blocks()
41
  with demo:
 
35
  plt.xlabel('SOURCE')
36
  plt.grid()
37
  metrics = {k: v for k, v in row.items() if k not in ['text', 'attention_maps', 'attention_maps_shape']}
38
+ return fig, metrics
39
 
40
  demo = gr.Blocks()
41
  with demo: