jaydeepkum's picture
Create new file
426ef32
raw history blame
No virus
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()