Update app.py

app.py

@@ -61,12 +61,13 @@ def analyze_sentence(index, vis_type, vis_format):
         fig_output = None
     else:
         fig_output = None
+        plot_data[np.arange(len(plot_data)), np.arange(len(plot_data))] = 0.
         top_values, top_indices = torch.tensor(plot_data).flatten().topk(30)
         topk_data = []
         for val, ind in zip(top_values, top_indices):
             ind = np.unravel_index(ind, plot_data.shape)
             topk_data += [str((tokenized[1+ind[0]], tokenized[1+ind[1]]))]
-        graph_output = '<div>' + '<br/>'.join(topk_data) + '</div>'
+        graph_output = '<div><b>' + text + '</b><br/>' + '<br/>'.join(topk_data) + '</div>'
     metrics = {'Metrics': 1}
     
     metrics.update({k: v for k, v in row.items() if k not in ['text', 'attention_maps', 'attention_maps_shape']})