Bases: HuggingFace
Wrapper for prompt injection detection model from Deepset.
For more information, please see the model card:
Source code in src/any_guardrail/guardrails/deepset/deepset.py
| class Deepset(HuggingFace):
"""Wrapper for prompt injection detection model from Deepset.
For more information, please see the model card:
- [Deepset](https://huggingface.co/deepset/deberta-v3-base-injection).
"""
SUPPORTED_MODELS: ClassVar = ["deepset/deberta-v3-base-injection"]
def _post_processing(self, model_outputs: dict[str, Any]) -> GuardrailOutput:
return _match_injection_label(model_outputs, DEEPSET_INJECTION_LABEL, self.model.config.id2label)
|