dar-tau commited on
Commit
7913cb8
·
verified ·
1 Parent(s): 872cb05

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -66,7 +66,7 @@ def analyze_sentence(index, vis_type, vis_format):
66
  for val, ind in zip(top_values, top_indices):
67
  ind = np.unravel_index(ind, plot_data.shape)
68
  topk_data += [(tokenized[1+ind[0]], tokenized[1+ind[1]])]
69
- graph_output = '<div>' + str(topk_data) + '</div>'
70
  metrics = {'Metrics': 1}
71
 
72
  metrics.update({k: v for k, v in row.items() if k not in ['text', 'attention_maps', 'attention_maps_shape']})
 
66
  for val, ind in zip(top_values, top_indices):
67
  ind = np.unravel_index(ind, plot_data.shape)
68
  topk_data += [(tokenized[1+ind[0]], tokenized[1+ind[1]])]
69
+ graph_output = '<div>' + '<br/>'.join(topk_data) + '</div>'
70
  metrics = {'Metrics': 1}
71
 
72
  metrics.update({k: v for k, v in row.items() if k not in ['text', 'attention_maps', 'attention_maps_shape']})