PreambleAI
/

prompt-injection-defense

Text Classification

Model card Files Files and versions

jeronimo9 commited on May 6, 2025

Commit

9913749

·

verified ·

1 Parent(s): cea7099

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -17,7 +17,7 @@ access_token = "ENTER_hf_API_KEY"
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 import torch
 tokenizer = AutoTokenizer.from_pretrained("answerdotai/ModernBERT-base")
-model = AutoModelForSequenceClassification.from_pretrained("PreambleAI/preamble-prompt-defense-v1", token=access_token)
 classifier = pipeline(
   "text-classification",
   model=model,

 from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 import torch
 tokenizer = AutoTokenizer.from_pretrained("answerdotai/ModernBERT-base")
+model = AutoModelForSequenceClassification.from_pretrained("PreambleAI/prompt-injection-defense", token=access_token)
 classifier = pipeline(
   "text-classification",
   model=model,