Views
No views yet

1from transformers import AutoModelForSequenceClassification, AutoTokenizer
2
3model = AutoModelForSequenceClassification.from_pretrained("hurtmongoose/Jailbreak-Detection-Models")
4tokenizer = AutoTokenizer.from_pretrained("hurtmongoose/Jailbreak-Detection-Models")
5
6inputs = tokenizer("This is a test jailbreak prompt", return_tensors="pt")
7outputs = model(**inputs)
8print(outputs.logits)
9📌 Notes
10
11Trained on jailbreak detection dataset
12
13Can be improved with more adversarial prompts