from transformers import pipeline

import gradio as gr


classifier = pipeline("ner", model=checkpoint)

examples = [
    "Patients présentant une sclérose en plaques rémittente-récurrente sévère et d’évolution rapide, définie par 2 poussées invalidantes ou plus au cours d’une année associées à 1 ou plusieurs lésion (s) rehaussée (s) après injection de Gadolinium sur l’IRM cérébrale ou une augmentation significative de la charge lésionnelle en T2 par rapport à une IRM antérieure récente.",
]

def ner(text):
    output = classifier(text)
    return {"text": text, "entities": output}

demo = gr.Interface(ner,
             gr.Textbox(placeholder="Enter sentence here..."),
             gr.HighlightedText(),
             examples=examples)

demo.launch()