522 Bytes
import gradio as gr | |
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="Patient is 50 yo M experienced a right bundle branch block (RBBB) yet tests for arrythmia found no results and heart health. Top goals."), outputs="highlightedtext") | |
interface.launch() |