diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md index 9004edd..c298068 100644 --- a/CONTRIBUTING.md +++ b/CONTRIBUTING.md @@ -18,7 +18,7 @@ uv sync --dev | Extra | Enables | |-------|---------| -| `ml` | ML span model (`Guard.with_tiny()`) | +| `ml` | ML span model (`Guard(model="tiny")`) | | `yara` | YARA code/SQLi/XSS scanner | | `presidio` | Presidio PII scanner | | `litellm` | LLM judge for borderline cases | diff --git a/assets/hf-org-card/README.md b/assets/hf-org-card/README.md index ce6c61e..13d837d 100644 --- a/assets/hf-org-card/README.md +++ b/assets/hf-org-card/README.md @@ -36,7 +36,7 @@ pip install "unplug-ai[ml]" ```python from unplug import Guard -guard = Guard.with_tiny() # auto-downloads unplug-tiny-v1 +guard = Guard(model="tiny",auto_download_model=True) # auto-downloads unplug-tiny-v1 result = guard.scan(untrusted_text) if not result.safe: use(result.redacted_text) # attack removed, content preserved