Update app.py

app.py

@@ -105,24 +105,11 @@ def create_mask_dict(entities):
                 mask_dict[entity['word']] = f"{entity['entity_group']}_{entity_counters[entity['entity_group']]}"
     return mask_dict
 
-def create_masked_text(input_text, entities):
-    mask_dict = create_mask_dict(entities)
-    
-    masked_text = input_text
-    for entity in sorted(entities, key=lambda x: x['start'], reverse=True):
-        if entity['entity_group'] not in ['CARDINAL', 'EVENT']:
-            masked_text = (
-                masked_text[:entity['start']] +
-                f"<{mask_dict[entity['word']]}> " +  # Use angle brackets for clarity
-                masked_text[entity['end']:]
-            )
-    return masked_text
-    
 def replace_words_in_text(input_text, entities):
     replace_dict = create_mask_dict(entities)
     for word, replacement in replace_dict.items():
-        text = text.replace(word, replacement)
-    return text
+        input_text = input_text.replace(word, replacement)
+    return input_text
 
 Run_Button = st.button("Run")