Add drop down model

app.py

@@ -25,8 +25,8 @@ def cached_anonimize_text(text, operator):
     return FlairRecognizer2.anonymize(text, operator)
 
 @st.cache_resource
-def anonymize(text, operator):
-    return PiiMaskingService().anonymize(text, operator)
+def anonymize(text, operator, model):
+    return PiiMaskingService().anonymize(text, operator, model)
 
 def download_masked_file(masked_text, file_extension):
     
@@ -91,16 +91,14 @@ def main():
             """,
     )
 
-    # st_model = st.sidebar.selectbox(
-    #     "NER model package",
-    #     [
-    #         "spaCy/en_core_web_lg",
-    #         "flair/ner-english-large",
-    #         "HuggingFace/obi/deid_roberta_i2b2",
-    #         "HuggingFace/StanfordAIMI/stanford-deidentifier-base",
-    #     ],
-    #     index=2,
-    # )
+    st_model = st.sidebar.selectbox(
+        "NER model package",
+        [
+            "flair/ner-english-large",
+            "HuggingFace/obi/deid_roberta_i2b2",
+        ],
+        index=1,
+    )
     
     masked_text_public = ''
     if upload_option == 'Text Input':
@@ -108,7 +106,7 @@ def main():
         if st.button('Analyze'):
             with st.spinner('Wait for it... the model is loading'):
                 # cached_predict_ner_tags(input_text)
-                masked_text = anonymize(input_text, st_operator)
+                masked_text = anonymize(input_text, st_operator, st_model)
                 # masked_text = cached_anonimize_text(input_text, st_operator)
             st.text_area("Masked text:", value=masked_text, height=200)
     elif upload_option == 'File Upload':