Spaces:

wandb
/

guardrails-genie

Running

ash0ts commited on Nov 22, 2024

Commit

41eea30

1 Parent(s): 0f0578b

Update presidio_pii_guardrail.py

Files changed (1) hide show

guardrails_genie/guardrails/pii/presidio_pii_guardrail.py CHANGED Viewed

@@ -14,6 +14,7 @@ class PresidioPIIGuardrailResponse(BaseModel):
     explanation: str
     anonymized_text: Optional[str] = None
 class PresidioPIIGuardrail(Guardrail):
     AVAILABLE_ENTITIES: ClassVar[List[str]] = [
         "PERSON", "EMAIL_ADDRESS", "PHONE_NUMBER", "LOCATION",

     explanation: str
     anonymized_text: Optional[str] = None
+#TODO: Add support for transformers workflow and not just Spacy
 class PresidioPIIGuardrail(Guardrail):
     AVAILABLE_ENTITIES: ClassVar[List[str]] = [
         "PERSON", "EMAIL_ADDRESS", "PHONE_NUMBER", "LOCATION",