protectai
/

deberta-v3-base-prompt-injection

Text Classification

prompt-injection

Generated from Trainer

Model card Files Files and versions

asofter commited on Dec 18, 2023

Commit

ca29d90

·

1 Parent(s): 36264af

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -94,7 +94,8 @@ Loading the model requires the [🤗 Optimum](https://huggingface.co/docs/optimu
 from optimum.onnxruntime import ORTModelForSequenceClassification
 from transformers import AutoTokenizer, pipeline
-tokenizer = AutoTokenizer.from_pretrained("laiyer/deberta-v3-base-prompt-injection")
 model = ORTModelForSequenceClassification.from_pretrained("laiyer/deberta-v3-base-prompt-injection", export=False, subfolder="onnx")
 classifier = pipeline(

 from optimum.onnxruntime import ORTModelForSequenceClassification
 from transformers import AutoTokenizer, pipeline
+tokenizer = AutoTokenizer.from_pretrained("laiyer/deberta-v3-base-prompt-injection", subfolder="onnx")
+tokenizer.model_input_names = ["input_ids", "attention_mask"]
 model = ORTModelForSequenceClassification.from_pretrained("laiyer/deberta-v3-base-prompt-injection", export=False, subfolder="onnx")
 classifier = pipeline(