Update flair_recognizer.py

flair_recognizer.py

@@ -84,7 +84,7 @@ class FlairRecognizer(EntityRecognizer):
     ]
 
     MODEL_LANGUAGES = {
-        "en": "beki/flair-pii-english",
+        "en": "beki/flair-pii-english-large",
     }
 
     PRESIDIO_EQUIVALENCES = {
@@ -219,26 +219,3 @@ class FlairRecognizer(EntityRecognizer):
         return any(
             [entity in egrp and label in lgrp for egrp, lgrp in check_label_groups]
         )
-
-
-if __name__ == "__main__":
-
-    from presidio_analyzer import AnalyzerEngine, RecognizerRegistry
-
-    flair_recognizer = (
-        FlairRecognizer()
-    )  # This would download a very large (+2GB) model on the first run
-
-    registry = RecognizerRegistry()
-    registry.add_recognizer(flair_recognizer)
-
-    analyzer = AnalyzerEngine(registry=registry)
-
-    results = analyzer.analyze(
-        "{first_name: Moustafa, sale_id: 235234}",
-        language="en",
-        return_decision_process=True,
-    )
-    for result in results:
-        print(result)
-        print(result.analysis_explanation)