Spaces:
Running
Running
petrsovadina
commited on
Commit
•
2c65ed9
1
Parent(s):
3aaa1e8
Update presidio_helpers.py
Browse files- 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 "
|
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 |
|