petrsovadina commited on
Commit
754824e
1 Parent(s): 3152804

Update presidio_helpers.py

Browse files
Files changed (1) hide show
  1. 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 "spacy" in model_family.lower():
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")