idanpers
/

JailBreakModel

Text Classification

Generated from Trainer

Model card Files Files and versions

idanpers commited on Nov 4, 2024

Commit

8aa027c

·

verified ·

1 Parent(s): e2b88be

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -56,8 +56,8 @@ To run the inference pipeline for classifying prompts, follow these steps:
   # Load model directly
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
-tokenizer = AutoTokenizer.from_pretrained("idanpers/electra-trainer")
-model = AutoModelForSequenceClassification.from_pretrained("idanpers/electra-trainer")
 use:

   # Load model directly
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
+tokenizer = AutoTokenizer.from_pretrained("idanpers/JailBreakModel")
+model = AutoModelForSequenceClassification.from_pretrained("idanpers/JailBreakModel")
 use: