Update README.md

README.md

@@ -71,8 +71,8 @@ The model's performance is dependent on the nature and quality of the training d
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 import torch
 
-tokenizer = AutoTokenizer.from_pretrained("laiyer/deberta-v3-base-prompt-injection")
-model = AutoModelForSequenceClassification.from_pretrained("laiyer/deberta-v3-base-prompt-injection")
+tokenizer = AutoTokenizer.from_pretrained("ProtectAI/deberta-v3-base-prompt-injection")
+model = AutoModelForSequenceClassification.from_pretrained("ProtectAI/deberta-v3-base-prompt-injection")
 
 classifier = pipeline(
   "text-classification",
@@ -94,9 +94,9 @@ 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", subfolder="onnx")
+tokenizer = AutoTokenizer.from_pretrained("ProtectAI/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")
+model = ORTModelForSequenceClassification.from_pretrained("ProtectAI/deberta-v3-base-prompt-injection", export=False, subfolder="onnx")
 
 classifier = pipeline(
   task="text-classification",
@@ -162,10 +162,10 @@ get help for package usage or contributions, or engage in discussions about LLM
 
 ```
 @misc{deberta-v3-base-prompt-injection,
-  author = {Laiyer.ai},
+  author = {ProtectAI.com},
   title = {Fine-Tuned DeBERTa-v3 for Prompt Injection Detection},
   year = {2023},
   publisher = {HuggingFace},
-  url = {https://huggingface.co/laiyer/deberta-v3-base-prompt-injection},
+  url = {https://huggingface.co/ProtectAI/deberta-v3-base-prompt-injection},
 }
 ```