Keyurjotaniya007's picture
Update app.py
0866bae verified
raw
history blame contribute delete
777 Bytes
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()