Spaces:
Runtime error
Runtime error
| import os | |
| #os.system('pip install https://huggingface.co/kormilitzin/en_core_med7_lg/resolve/main/en_core_med7_lg-any-py3-none-any.whl') | |
| os.system('pip install https://huggingface.co/kormilitzin/en_core_med7_trf/resolve/main/en_core_med7_trf-any-py3-none-any.whl') | |
| # Using spacy.load(). | |
| #import spacy | |
| #nlp = spacy.load("en_core_med7_trf") | |
| # Importing as module. | |
| #import en_core_med7_trf | |
| #nlp = en_core_med7_trf.load()') | |
| import gradio as gr | |
| from spacy import displacy | |
| import spacy | |
| med7 = spacy.load("en_core_med7_trf") | |
| def get_med7_ent(text): | |
| # create distinct colours for labels | |
| col_dict = {} | |
| seven_colours = ['#e6194B', '#3cb44b', '#ffe119', '#ffd8b1', '#f58231', '#f032e6', '#42d4f4'] | |
| for label, colour in zip(med7.pipe_labels['ner'], seven_colours): | |
| col_dict[label] = colour | |
| options = {'ents': med7.pipe_labels['ner'], 'colors':col_dict} | |
| doc = med7(text) | |
| html = displacy.render(doc, style='ent',options=options) | |
| return html | |
| exp=["A patient was prescribed Magnesium hydroxide 400mg/5ml suspension PO of total 30ml bid for the next 5 days."] | |
| desc="Med7 — An information extraction model for clinical natural language processing. More information about the model development can be found in recent pre-print: Med7: a transferable clinical natural language processing model for electronic health records." | |
| inp=gr.inputs.Textbox(lines=5, placeholder=None, default="", label="Text") | |
| out=gr.outputs.HTML(label=None) | |
| iface = gr.Interface(fn=get_med7_ent, inputs=inp, outputs=out,examples=exp,article=desc,title="Med7",theme="huggingface",layout='horizontal') | |
| iface.launch() |