Spaces:
Running
Running
petrsovadina
commited on
Commit
•
754824e
1
Parent(s):
3152804
Update presidio_helpers.py
Browse files- presidio_helpers.py +2 -8
presidio_helpers.py
CHANGED
@@ -3,7 +3,6 @@ Helper methods for the Presidio Streamlit app
|
|
3 |
"""
|
4 |
from typing import List, Optional, Tuple
|
5 |
import logging
|
6 |
-
import os
|
7 |
import streamlit as st
|
8 |
from presidio_analyzer import (
|
9 |
AnalyzerEngine,
|
@@ -21,10 +20,7 @@ from openai_fake_data_generator import (
|
|
21 |
OpenAIParams,
|
22 |
create_prompt,
|
23 |
)
|
24 |
-
from presidio_nlp_engine_config import
|
25 |
-
create_nlp_engine_with_spacy,
|
26 |
-
create_nlp_engine_with_transformers,
|
27 |
-
)
|
28 |
|
29 |
logger = logging.getLogger("presidio-streamlit")
|
30 |
|
@@ -36,9 +32,7 @@ def nlp_engine_and_registry(
|
|
36 |
ta_endpoint: Optional[str] = None,
|
37 |
) -> Tuple[NlpEngine, RecognizerRegistry]:
|
38 |
"""Create the NLP Engine instance based on the requested model."""
|
39 |
-
if "
|
40 |
-
return create_nlp_engine_with_spacy(model_path)
|
41 |
-
elif "transformers" in model_family.lower() or "iiiorg" in model_family.lower():
|
42 |
return create_nlp_engine_with_transformers(model_path)
|
43 |
else:
|
44 |
raise ValueError(f"Model family {model_family} not supported")
|
|
|
3 |
"""
|
4 |
from typing import List, Optional, Tuple
|
5 |
import logging
|
|
|
6 |
import streamlit as st
|
7 |
from presidio_analyzer import (
|
8 |
AnalyzerEngine,
|
|
|
20 |
OpenAIParams,
|
21 |
create_prompt,
|
22 |
)
|
23 |
+
from presidio_nlp_engine_config import create_nlp_engine_with_transformers
|
|
|
|
|
|
|
24 |
|
25 |
logger = logging.getLogger("presidio-streamlit")
|
26 |
|
|
|
32 |
ta_endpoint: Optional[str] = None,
|
33 |
) -> Tuple[NlpEngine, RecognizerRegistry]:
|
34 |
"""Create the NLP Engine instance based on the requested model."""
|
35 |
+
if "transformers" in model_family.lower() or "iiiorg" in model_family.lower():
|
|
|
|
|
36 |
return create_nlp_engine_with_transformers(model_path)
|
37 |
else:
|
38 |
raise ValueError(f"Model family {model_family} not supported")
|