Text Classification
Transformers
Safetensors
English
distilbert
prompt-injection-detection
jailbreak-detection
security
text-embeddings-inference
Instructions to use KIMMISEON/distilbert-prompt-guard-4class with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use KIMMISEON/distilbert-prompt-guard-4class with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="KIMMISEON/distilbert-prompt-guard-4class")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("KIMMISEON/distilbert-prompt-guard-4class") model = AutoModelForSequenceClassification.from_pretrained("KIMMISEON/distilbert-prompt-guard-4class") - Notebooks
- Google Colab
- Kaggle
| { | |
| "label_names": [ | |
| "benign", | |
| "prompt_injection", | |
| "jailbreak", | |
| "sensitive_access" | |
| ], | |
| "label_to_id": { | |
| "benign": 0, | |
| "prompt_injection": 1, | |
| "jailbreak": 2, | |
| "sensitive_access": 3 | |
| }, | |
| "id_to_label": { | |
| "0": "benign", | |
| "1": "prompt_injection", | |
| "2": "jailbreak", | |
| "3": "sensitive_access" | |
| } | |
| } |