Update app.py

app.py

@@ -65,6 +65,8 @@ def analyze_sentence(index, vis_type, vis_format):
         top_values, top_indices = torch.tensor(plot_data).flatten().topk(30)
         topk_data = []
         for val, ind in zip(top_values, top_indices):
+            if val < 0.5:
+                break
             ind = np.unravel_index(ind, plot_data.shape)
             topk_data += [str((tokenized[1+ind[0]], tokenized[1+ind[1]]))]
         graph_output = '<div><b>' + text + '</b><br/>' + '<br/>'.join(topk_data) + '</div>'