Update analyze.py

analyze.py

@@ -46,11 +46,10 @@ def batched(
 
 
 def mask(text: str) -> str:
-    return text  # don't apply mask for demo
-    # return " ".join(
-    #     word[: min(2, len(word) - 1)] + re.sub("[A-Za-z0-9]", "*", word[min(2, len(word) - 1) :])
-    #     for word in text.split(" ")
-    # )
+    return " ".join(
+        word[: min(2, len(word) - 1)] + re.sub("[A-Za-z0-9]", "*", word[min(2, len(word) - 1) :])
+        for word in text.split(" ")
+    )
 
 
 def get_strings(row_content: Any) -> str:
@@ -101,7 +100,7 @@ def analyze(
     ]
     return [
         PresidioEntity(
-            text=mask(texts[i * len(scanned_columns) + j][recognizer_result.start : recognizer_result.end]),
+            text=texts[i * len(scanned_columns) + j][recognizer_result.start : recognizer_result.end],
             type=recognizer_result.entity_type,
             row_idx=row_idx,
             column_name=column_name,