petrsovadina commited on
Commit
2c65ed9
1 Parent(s): 3aaa1e8

Update presidio_helpers.py

Browse files
Files changed (1) hide show
  1. presidio_helpers.py +2 -1
presidio_helpers.py CHANGED
@@ -34,7 +34,7 @@ def nlp_engine_and_registry(
34
  ta_endpoint: Optional[str] = None,
35
  ) -> Tuple[NlpEngine, RecognizerRegistry]:
36
  """Create the NLP Engine instance based on the requested model."""
37
- if "transformers" in model_family.lower() or "iiiorg" in model_family.lower():
38
  return create_nlp_engine_with_transformers(model_path)
39
  else:
40
  raise ValueError(f"Model family {model_family} not supported")
@@ -153,6 +153,7 @@ def create_fake_data(
153
  return "Please provide your OpenAI key"
154
  results = anonymize(text=text, operator="replace", analyze_results=analyze_results)
155
  prompt = create_prompt(results.text)
 
156
  fake = call_completion_model(prompt=prompt, openai_params=openai_params)
157
  return fake
158
 
 
34
  ta_endpoint: Optional[str] = None,
35
  ) -> Tuple[NlpEngine, RecognizerRegistry]:
36
  """Create the NLP Engine instance based on the requested model."""
37
+ if "iiiorg" in model_family.lower() or "transformers" in model_family.lower():
38
  return create_nlp_engine_with_transformers(model_path)
39
  else:
40
  raise ValueError(f"Model family {model_family} not supported")
 
153
  return "Please provide your OpenAI key"
154
  results = anonymize(text=text, operator="replace", analyze_results=analyze_results)
155
  prompt = create_prompt(results.text)
156
+ print(f"Prompt: {prompt}")
157
  fake = call_completion_model(prompt=prompt, openai_params=openai_params)
158
  return fake
159