* choose recognizers

prompt.py

@@ -19,7 +19,7 @@ from llm_guard.input_scanners import (
     Toxicity,
 )
 from llm_guard.input_scanners.anonymize import default_entity_types
-from llm_guard.input_scanners.anonymize_helpers.analyzer import allowed_recognizers
+from llm_guard.input_scanners.anonymize_helpers.analyzer import RECOGNIZER_SPACY_EN_PII_DISTILBERT, RECOGNIZER_SPACY_EN_PII_FAST
 from llm_guard.vault import Vault
 
 logger = logging.getLogger("llm-guard-playground")
@@ -103,7 +103,7 @@ def init_settings() -> (List, Dict):
             )
             st_anon_recognizer = st.selectbox(
                 "Recognizer",
-                allowed_recognizers,
+                [RECOGNIZER_SPACY_EN_PII_DISTILBERT, RECOGNIZER_SPACY_EN_PII_FAST],
                 index=1,
             )
 

requirements.txt

@@ -3,3 +3,4 @@ llm-guard==0.3.0
 pandas==2.1.0
 streamlit==1.27.2
 streamlit-tags==1.2.8
+https://huggingface.co/beki/en_spacy_pii_fast/resolve/main/en_spacy_pii_fast-any-py3-none-any.whl