Update app.py

app.py

@@ -2,29 +2,25 @@ 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",
-    aggregation_strategy="simple"
+    grouped_entities=True   
 )
 
 def extract_entities(text):
     results = ner(text)
-    entities = [{"word": r["word"], "label": r.get("entity_group"), "score": float(r.get("score", 0))} for r in results]
-
-    annotated = "  |  ".join([f"{e['word']} ({e['label']})" for e in entities]) if entities else "No entities found."
-
-    return entities, annotated
+    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...", lines=3)
-        out_json = gr.JSON(label="Entities (JSON)")
-        out_text = gr.Textbox(label="Annotated Text")
+        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_json, out_text])
+    btn.click(fn=extract_entities, inputs=inp, outputs=out)
 
 if __name__ == "__main__":
     demo.launch()