Hugging Face's logo

Spaces:
wandb
/
guardrails-genie
Runtime error

App Files Community
1
ash0ts commited on
Commit
41eea30
·
1 Parent(s): 0f0578b

Update presidio_pii_guardrail.py

Browse files
Files changed (1) hide show
  1. guardrails_genie/guardrails/pii/presidio_pii_guardrail.py +1 -0
guardrails_genie/guardrails/pii/presidio_pii_guardrail.py CHANGED
@@ -14,6 +14,7 @@ class PresidioPIIGuardrailResponse(BaseModel):
14
  explanation: str
15
  anonymized_text: Optional[str] = None
16
 
 
17
  class PresidioPIIGuardrail(Guardrail):
18
  AVAILABLE_ENTITIES: ClassVar[List[str]] = [
19
  "PERSON", "EMAIL_ADDRESS", "PHONE_NUMBER", "LOCATION",
 
14
  explanation: str
15
  anonymized_text: Optional[str] = None
16
 
17
+ #TODO: Add support for transformers workflow and not just Spacy
18
  class PresidioPIIGuardrail(Guardrail):
19
  AVAILABLE_ENTITIES: ClassVar[List[str]] = [
20
  "PERSON", "EMAIL_ADDRESS", "PHONE_NUMBER", "LOCATION",