Update app.py

app.py

@@ -44,8 +44,8 @@ def analyze_sentence(index, vis_type, vis_format):
         fig_output, graph_output = fig, ""
     else:
         ex = [{
-            "words": [{"text": x, "tag": "N/A"} for x in tokenized[1:]],
-            "arcs": [{"start": j, "end": i, "label": "{:.2f}".format(plot_data[i, j]),
+            "words": [{"text": x, "tag": ""} for x in tokenized[1:]],
+            "arcs": [{"start": j, "end": i, "label": "",
                       "dir": "right"} 
                      for i in range(len(tokenized)-1) for j in range(i) if plot_data[i, j] > 0.5
                     ]
@@ -70,7 +70,7 @@ with demo:
     vis_format_checkbox = gr.Radio(['Plot', 'Graph'])
         
     output = gr.Plot(label="Plot", container=True)
-    graph_output = gr.HTML(label="Graph")
+    graph_output = gr.HTML()
     metrics = gr.Label("Metrics")
     btn.click(analyze_sentence, [sentence_dropdown, vis_dropdown, vis_format_checkbox], [output, graph_output, metrics])