vijil
/

mbert-prompt-injection

Model card Files Files and versions Community

vijilpd commited on 4 days ago

Commit

4fc3ef8

·

verified ·

1 Parent(s): 1d3c053

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -25,7 +25,7 @@ The vijil/mbert-prompt-injection model is designed to enhance security in langua
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 import torch
-tokenizer = AutoTokenizer.from_pretrained("vijil/mbert-prompt-injection")
 model = AutoModelForSequenceClassification.from_pretrained("vijil/mbert-prompt-injection")
 classifier = pipeline(

 from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 import torch
+tokenizer = AutoTokenizer.from_pretrained("answerdotai/ModernBERT-base")
 model = AutoModelForSequenceClassification.from_pretrained("vijil/mbert-prompt-injection")
 classifier = pipeline(