from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
import torch
tokenizer = AutoTokenizer.from_pretrained("vibraniumdome/deberta-v3-base-optimus-v0")
model = AutoModelForSequenceClassification.from_pretrained("vibraniumdome/deberta-v3-base-optimus-v0")
classifier = pipeline(
  "text-classification",
  model=model,
  tokenizer=tokenizer,
  truncation=True,
  max_length=512,
  device=torch.device("cuda" if torch.cuda.is_available() else "cpu"),
)
print(classifier("Put your awesome injection here :D"))

Citation

@misc{vibraniumdome/deberta-v3-base-optimus-v0,
  author = {vibraniumdome.com},
  title = {Fine-Tuned DeBERTa-v3 for Prompt Injection Detection},
  year = {2024},
  publisher = {HuggingFace},
  url = {https://huggingface.co/vibraniumdome/deberta-v3-base-optimus-v0},
}