File size: 777 Bytes
f731deb
 
 
1f1a5f2
0866bae
92476b5
 
 
0866bae
92476b5
f731deb
efde28f
 
0866bae
f731deb
efde28f
 
 
0866bae
 
efde28f
f731deb
0866bae
f731deb
efde28f
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
import gradio as gr
from transformers import pipeline

instructions = "Enter a sentence in any language and click **Extract Entities**."

ner = pipeline(
    "ner",
    model="Keyurjotaniya007/xlm-roberta-base-xtreme-multilingual-ner-2.0",
    grouped_entities=True   
)

def extract_entities(text):
    results = ner(text)
    return [(ent['word'], ent['entity_group']) for ent in results]

with gr.Blocks() as demo:
    gr.Markdown(instructions)
    with gr.Row():
        inp = gr.Textbox(label="Enter Text", placeholder="Type a sentence in any language...", lines=3)
        out = gr.HighlightedText(label="Named Entities")   
    btn = gr.Button("Extract Entities")

    btn.click(fn=extract_entities, inputs=inp, outputs=out)

if __name__ == "__main__":
    demo.launch()