Spaces:

protectai
/

llm-guard-playground

Running

asofter commited on Oct 16, 2023

Commit

8950a27

•

1 Parent(s): 26c618b

* choose recognizers

Files changed (2) hide show

prompt.py CHANGED Viewed

@@ -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.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,
                 index=1,
             )

     Toxicity,
 )
 from llm_guard.input_scanners.anonymize import default_entity_types
+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")
             )
             st_anon_recognizer = st.selectbox(
                 "Recognizer",
+                [RECOGNIZER_SPACY_EN_PII_DISTILBERT, RECOGNIZER_SPACY_EN_PII_FAST],
                 index=1,
             )

requirements.txt CHANGED Viewed

@@ -3,3 +3,4 @@ llm-guard==0.3.0
 pandas==2.1.0
 streamlit==1.27.2
 streamlit-tags==1.2.8

 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