adit94 commited on
Commit
74235b1
·
verified ·
1 Parent(s): d6d4433

Update pii.py

Browse files
Files changed (1) hide show
  1. pii.py +2 -2
pii.py CHANGED
@@ -59,7 +59,7 @@ class FlairRecognizer(EntityRecognizer):
59
 
60
  def __init__(
61
  self,
62
- supported_language: str = "en",
63
  supported_entities: Optional[List[str]] = None,
64
  check_label_groups: Optional[Tuple[Set, Set]] = None,
65
  model: SequenceTagger = None,
@@ -218,7 +218,7 @@ class PII_IDENTIFIER:
218
  flair_recognizer = FlairRecognizer(supported_language="de")
219
  print("Flair loaded")
220
 
221
- registry = RecognizerRegistry(supported_language="de")
222
  #registry.load_predefined_recognizers()
223
  #registry.add_recognizer(SpacyRecognizer(supported_language="de"))
224
  #registry.add_recognizer(SpacyRecognizer(supported_language="en"))
 
59
 
60
  def __init__(
61
  self,
62
+ supported_language: str = "de",
63
  supported_entities: Optional[List[str]] = None,
64
  check_label_groups: Optional[Tuple[Set, Set]] = None,
65
  model: SequenceTagger = None,
 
218
  flair_recognizer = FlairRecognizer(supported_language="de")
219
  print("Flair loaded")
220
 
221
+ registry = RecognizerRegistry()
222
  #registry.load_predefined_recognizers()
223
  #registry.add_recognizer(SpacyRecognizer(supported_language="de"))
224
  #registry.add_recognizer(SpacyRecognizer(supported_language="en"))