Update flair_recognizer.py

flair_recognizer.py

@@ -84,7 +84,8 @@ class FlairRecognizer(EntityRecognizer):
     ]
 
     MODEL_LANGUAGES = {
-        "en": "beki/flair-pii-english",
+        "en":"beki/flair-pii-english-large",
+        # "en":"flair-trf.pt",
     }
 
     PRESIDIO_EQUIVALENCES = {
@@ -219,3 +220,26 @@ 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)