update app

app.py

@@ -34,7 +34,7 @@ def extract_entities(sentence):
     for key in results.keys():
         entity_results.append(print_nicely(results[key]))
 
-    return "\n".join(entity_results)
+    return results
 
 
 # Create Gradio interface
@@ -42,7 +42,7 @@ def ner_app_interface():
     input_sentence = gr.Textbox(
         lines=5, label="Input Sentence", placeholder="Enter a sentence for NER..."
     )
-    output_entities = gr.Textbox(label="Extracted Entities")
+    output_entities = gr.HighlightedText(label="Extracted Entities")
 
     # Interface definition
     interface = gr.Interface(