WIP

app.py

@@ -5,11 +5,8 @@ import gradio as gr
 
 def ner(text):
     api = gr.Interface.load("d4data/biomedical-ner-all", src="models")
-    output = api(text)
-    replaced_spans = [
-        (key, None) if value == "No Disease" else (key, value) for (key, value) in spans
-    ]
-    return replaced_spans
+    result = api(text)
+    return result
 
 
 EXAMPLE_TEXTS = []
@@ -21,7 +18,7 @@ with open("examples.json", "r") as f:
 interface = gr.Interface(
     ner,
     inputs=gr.Textbox(label="Input", value=""),
-    outputs="highlightedtext",
+    outputs=["highlighttext", "json"],
     examples=EXAMPLE_TEXTS,
 )