Update app.py

app.py

@@ -59,6 +59,7 @@ def ner(text):
   output = my_classifier(text)
   for entity in output:
     entity['entity'] = entity.pop('entity_group')
+      
   return {'text': text, 'entities': output}, text
 
 examples = ['tôi cần thuê à tôi muốn bay một chuyến khứ_hồi từ đà_nẵng đến đà_lạt', 
@@ -69,7 +70,7 @@ examples = ['tôi cần thuê à tôi muốn bay một chuyến khứ_hồi từ
 
 demo = gr.Interface(ner, 
                     gr.Textbox(label='Text', placeholder="Enter sentence here..."),
-                    outputs=[gr.HighlightedText(label='Highlighted Output'), 'text'],
+                    outputs=[gr.HighlightedText(label='Highlighted Output'), gr.Textbox(label='RDR-Segmented sentence(Preprocessed before input to model)')],
                     examples=examples,
                     title="Disfluency Detection",
                     description="This is an easy-to-use built in Gradio for desmontrating a NER System that identifies disfluency-entities in \