Update app.py (#1)

- Update app.py (b116a1d49bbd73cd026108c68ac5c4247e1c4f50)


Co-authored-by: Abubakar Abid <abidlabs@users.noreply.huggingface.co>

app.py

@@ -1,7 +1,13 @@
 import gradio as gr
 
-gr.Interface.load("ugaray96/biobert_ncbi_disease_ner" \
-#gr.Interface.load("huggingface/huggingface-course/bert-finetuned-ner" \
-,src='models', title=None, inputs=gr.Textbox(label="Input", value="Autophagy maintains tumour growth through circulating arginine. Acute, \
+def ner(text):
+    api = gr.Interface.load("ugaray96/biobert_ncbi_disease_ner", src='models')
+    spans = api(text)
+    replaced_spans = [(key, None) if value=='No Disease' else (key, value) for (key, value) in spans]
+    return replaced_spans
+
+interface = gr.Interface(ner, inputs=gr.Textbox(label="Input", value="Autophagy maintains tumour growth through circulating arginine. Acute, \
 whole-body deletion of the essential autophagy gene Atg7 in adult mice causes a systemic metabolic defect that manifests as starvation  \
-intolerance and gradual loss of white adipose tissue, liver glycogen and muscle mass. Cancer cells also benefit from autophagy.")).launch()
+intolerance and gradual loss of white adipose tissue, liver glycogen and muscle mass. Cancer cells also benefit from autophagy."), outputs="highlightedtext")
+
+interface.launch()