Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -29,16 +29,15 @@ st.sidebar.header("PHI De-identification with Presidio")
|
|
29 |
|
30 |
model_help_text = "Select Named Entity Recognition (NER) model for PHI detection."
|
31 |
model_list = [
|
32 |
-
("spaCy/en_core_web_lg", "https://huggingface.co/spacy/en_core_web_lg"),
|
33 |
-
("HuggingFace/obi/deid_roberta_i2b2", "https://huggingface.co/obi/deid_roberta_i2b2"),
|
34 |
("flair/ner-english-large", "https://huggingface.co/flair/ner-english-large"),
|
|
|
35 |
("HuggingFace/StanfordAIMI/stanford-deidentifier-base", "https://huggingface.co/StanfordAIMI/stanford-deidentifier-base"),
|
36 |
]
|
37 |
|
38 |
st_model = st.sidebar.selectbox(
|
39 |
"NER model package",
|
40 |
[model[0] for model in model_list],
|
41 |
-
index=
|
42 |
help=model_help_text,
|
43 |
)
|
44 |
|
@@ -48,7 +47,7 @@ st.sidebar.markdown(f"[View model on HuggingFace]({selected_model_url})")
|
|
48 |
|
49 |
# Extract model package
|
50 |
st_model_package = st_model.split("/")[0]
|
51 |
-
st_model = st_model if st_model_package.lower() not in ("
|
52 |
|
53 |
analyzer_params = (st_model_package, st_model)
|
54 |
st.sidebar.warning("Note: Models might take some time to download on first run.")
|
@@ -97,7 +96,7 @@ with col1:
|
|
97 |
analyzer = analyzer_engine(*analyzer_params)
|
98 |
except Exception as e:
|
99 |
st.error(f"Failed to load model: {str(e)}")
|
100 |
-
st.info("Ensure models are downloaded
|
101 |
raise
|
102 |
|
103 |
# Analyze
|
|
|
29 |
|
30 |
model_help_text = "Select Named Entity Recognition (NER) model for PHI detection."
|
31 |
model_list = [
|
|
|
|
|
32 |
("flair/ner-english-large", "https://huggingface.co/flair/ner-english-large"),
|
33 |
+
("HuggingFace/obi/deid_roberta_i2b2", "https://huggingface.co/obi/deid_roberta_i2b2"),
|
34 |
("HuggingFace/StanfordAIMI/stanford-deidentifier-base", "https://huggingface.co/StanfordAIMI/stanford-deidentifier-base"),
|
35 |
]
|
36 |
|
37 |
st_model = st.sidebar.selectbox(
|
38 |
"NER model package",
|
39 |
[model[0] for model in model_list],
|
40 |
+
index=0,
|
41 |
help=model_help_text,
|
42 |
)
|
43 |
|
|
|
47 |
|
48 |
# Extract model package
|
49 |
st_model_package = st_model.split("/")[0]
|
50 |
+
st_model = st_model if st_model_package.lower() not in ("huggingface") else "/".join(st_model.split("/")[1:])
|
51 |
|
52 |
analyzer_params = (st_model_package, st_model)
|
53 |
st.sidebar.warning("Note: Models might take some time to download on first run.")
|
|
|
96 |
analyzer = analyzer_engine(*analyzer_params)
|
97 |
except Exception as e:
|
98 |
st.error(f"Failed to load model: {str(e)}")
|
99 |
+
st.info("Ensure models are downloaded and check network/permissions.")
|
100 |
raise
|
101 |
|
102 |
# Analyze
|