Spaces:
Sleeping
Sleeping
Update utils/masker3.py
Browse files- 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 |
|