Update app.py

app.py

@@ -173,7 +173,7 @@ if Run_Button and input_text:
             entity['masked_word'] = entity['word']
     
     df = pd.DataFrame.from_dict(output_comb)
-    cols_to_keep = ['word', 'masked_word', 'entity_group', 'score', 'start', 'end']
+    cols_to_keep = ['word', 'word', 'entity_group', 'score', 'start', 'end']
     df_final = df[cols_to_keep]
     
     st.subheader("Recognized Entities")