Sparkonix commited on
Commit
7030c07
·
1 Parent(s): 7f4c4f8

change entity type text

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -314,7 +314,7 @@ class PIIMasker:
314
  new_text_parts.append(text[current_pos:entity.start])
315
 
316
  # Add the mask
317
- mask = f"[{entity.entity_type.upper()}]" # Changed to upper for clarity
318
  new_text_parts.append(mask)
319
 
320
  current_pos = entity.end
 
314
  new_text_parts.append(text[current_pos:entity.start])
315
 
316
  # Add the mask
317
+ mask = f"[{entity.entity_type}]" # Changed to upper for clarity
318
  new_text_parts.append(mask)
319
 
320
  current_pos = entity.end