format trees

app.py

@@ -1,7 +1,11 @@
+import matplotlib
+matplotlib.use('Agg')
+
 import gradio
 import benepar
 import spacy
 import nltk
+from nltk.tree import Tree
 nltk.download('stopwords')
 
 from huggingface_hub import hf_hub_url, cached_download
@@ -37,7 +41,7 @@ def predict(sentence, model):
     elif model == "inside-outside":
         best_parse = Predictor(sentence=sentence).obtain_best_parse(predict_type="inside_outside", model=inside_outside_model, scale_axis=1, predict_batch_size=128)
     sentence_f1 = calculate_F1_for_spans(tree_to_spans(gold_standard), tree_to_spans(best_parse))
-    return gold_standard, best_parse, f"{sentence_f1:.2f}"
+    return Tree.fromstring(gold_standard).draw(), Tree.fromstring(best_parse).draw(), f"{sentence_f1:.2f}"
 
 
 iface = gradio.Interface(
@@ -64,4 +68,4 @@ iface = gradio.Interface(
         ["Bitcoin community under pressure to cut energy use .", "inside"],
     ],
 )
-iface.launch()
+iface.launch(share=True)

packages.txt

@@ -0,0 +1,2 @@
+ghostscript
+python3-tk