import gradio as gr
from transformers import pipeline
from transformers import  AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("51la5/roberta-large-NER")
model = AutoModelForTokenClassification.from_pretrained("51la5/roberta-large-NER")
classifier = pipeline("ner", model=model, tokenizer=tokenizer,grouped_entities=True)

def get_ner(text):
    output = classifier(text)
    for elm in output:
        elm['entity'] = elm['entity_group']
    return {"text": text, "entities": output}


demo = gr.Interface(fn=get_ner,
                    title="Atoqli nomlarni topish(NER)",
                    inputs=gr.Textbox(lines=4, placeholder="Matinni kiriting!", label="Matn*"),
                    outputs=gr.HighlightedText(label="Natija:")
                    )

demo.launch()