Update presidio_streamlit.py

presidio_streamlit.py

@@ -53,12 +53,7 @@ st_ta_key = st_ta_endpoint = ""
 
 model_list = [
     "spaCy/en_core_web_lg",
-    "flair/ner-english-large",
-    "HuggingFace/obi/deid_roberta_i2b2",
-    "HuggingFace/StanfordAIMI/stanford-deidentifier-base",
-    "stanza/en",
-    "Azure AI Language",
-    "Other",
+    "iiiorg/piiranha-v1-detect-personal-information",
 ]
 if not allow_other_models:
     model_list.pop()
@@ -76,25 +71,17 @@ st_model_package = st_model.split("/")[0]
 # Remove package prefix (if needed)
 st_model = (
     st_model
-    if st_model_package.lower() not in ("spacy", "stanza", "huggingface")
+    if st_model_package.lower() not in ("spacy","piiiranha")
     else "/".join(st_model.split("/")[1:])
 )
 
 if st_model == "Other":
     st_model_package = st.sidebar.selectbox(
-        "NER model OSS package", options=["spaCy", "stanza", "Flair", "HuggingFace"]
+        "NER model OSS package", options=["spacy","piiiranha"]
     )
     st_model = st.sidebar.text_input(f"NER model name", value="")
 
-if st_model == "Azure AI Language":
-    st_ta_key = st.sidebar.text_input(
-        f"Azure AI Language key", value=os.getenv("TA_KEY", ""), type="password"
-    )
-    st_ta_endpoint = st.sidebar.text_input(
-        f"Azure AI Language endpoint",
-        value=os.getenv("TA_ENDPOINT", default=""),
-        help="For more info: https://learn.microsoft.com/en-us/azure/cognitive-services/language-service/personally-identifiable-information/overview",  # noqa: E501
-    )
+
 
 
 st.sidebar.warning("Note: Models might take some time to download. ")