Update app.py

app.py

@@ -1,15 +1,12 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load the multilingual NER pipeline
-ner = pipeline("ner", model="Davlan/xlm-roberta-base-ner-hrl", grouped_entities=True)
+ner = pipeline("ner", model="Keyurjotaniya007/xlm-roberta-base-xtreme-multilingual-ner-2.0", grouped_entities=True)
 
-# Inference function
 def extract_entities(text):
     results = ner(text)
     return [(ent['word'], ent['entity_group']) for ent in results]
 
-# Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown(instructions)
     with gr.Row():
@@ -19,6 +16,5 @@ with gr.Blocks() as demo:
 
     btn.click(fn=extract_entities, inputs=inp, outputs=out)
 
-# Launch
 if __name__ == "__main__":
     demo.launch()