Update app.py

app.py

@@ -151,10 +151,13 @@ if Run_Button and input_text:
         all_outputs.extend(output)
     
     # Combine entities
+    
     output_comb = entity_comb(all_outputs)
     
     # Create mask dictionary
     mask_dict = create_mask_dict(output_comb)
+
+    masked_text = " ".join(entity['masked_word'] for entity in output_comb)
     
     # Apply masking and add masked_word column
     for entity in output_comb: