VGreatVig07 commited on
Commit
0e13165
·
verified ·
1 Parent(s): 063bb6f

Update utils/masker3.py

Browse files
Files changed (1) hide show
  1. utils/masker3.py +4 -1
utils/masker3.py CHANGED
@@ -70,6 +70,7 @@ def mask_pii(text: str) -> Dict[str, Any]:
70
 
71
  # Optional: Set category based on simple rule or ML model
72
  category = "Problem"
 
73
 
74
 
75
  # if lang == 'en':
@@ -86,7 +87,9 @@ def mask_pii(text: str) -> Dict[str, Any]:
86
  "input_email_body": text2,
87
  "list_of_masked_entities": sorted(entities, key=lambda x: x["position"][0]),
88
  "masked_email": masked_text,
89
- "category_of_the_email": category
 
 
90
  }
91
 
92
 
 
70
 
71
  # Optional: Set category based on simple rule or ML model
72
  category = "Problem"
73
+ eng_mask = masked_text
74
 
75
 
76
  # if lang == 'en':
 
87
  "input_email_body": text2,
88
  "list_of_masked_entities": sorted(entities, key=lambda x: x["position"][0]),
89
  "masked_email": masked_text,
90
+ "category_of_the_email": category,
91
+ "English_masked": eng_mask
92
+
93
  }
94
 
95