Hugging Face's logo

Solegon
/
prompt-safety-bert

Text Classification
Transformers
Safetensors
English
distilbert
safety
jailbreak-detection
prompt-injection
content-moderation
Eval Results (legacy)
text-embeddings-inference
Model card Files
xet
 Community
How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-classification", model="Solegon/prompt-safety-bert")
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("Solegon/prompt-safety-bert")
model = AutoModelForSequenceClassification.from_pretrained("Solegon/prompt-safety-bert")
Quick Links

🛡️ Prompt Safety BERT

A fine-tuned DistilBERT model for safe/unsafe prompt classification.

Downloads last month
4
Safetensors
Model size
67M params
Tensor type
F32
·
Inference Providers NEW
Text Classification
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Datasets used to train Solegon/prompt-safety-bert

Evaluation results