Update README.md

README.md

@@ -40,7 +40,7 @@ It achieves the following results on the evaluation set:
 
 ## Model details
 
-- **Fine-tuned by:** Laiyer.ai
+- **Fine-tuned by:** ProtectAI.com
 - **Model type:** distilroberta-base
 - **Language(s) (NLP):** English
 - **License:** Apache license 2.0
@@ -62,8 +62,8 @@ Additionally, `distilroberta-base` is case-sensitive model.
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 
-tokenizer = AutoTokenizer.from_pretrained("laiyer/distilroberta-base-rejection-v1")
-model = AutoModelForSequenceClassification.from_pretrained("laiyer/distilroberta-base-rejection-v1")
+tokenizer = AutoTokenizer.from_pretrained("ProtectAI/distilroberta-base-rejection-v1")
+model = AutoModelForSequenceClassification.from_pretrained("ProtectAI/distilroberta-base-rejection-v1")
 
 classifier = pipeline(
   "text-classification",
@@ -85,8 +85,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/distilroberta-base-rejection-v1", subfolder="onnx")
-model = ORTModelForSequenceClassification.from_pretrained("laiyer/distilroberta-base-rejection-v1", export=False, subfolder="onnx")
+tokenizer = AutoTokenizer.from_pretrained("ProtectAI/distilroberta-base-rejection-v1", subfolder="onnx")
+model = ORTModelForSequenceClassification.from_pretrained("ProtectAI/distilroberta-base-rejection-v1", export=False, subfolder="onnx")
 
 classifier = pipeline(
   task="text-classification",
@@ -152,10 +152,10 @@ get help for package usage or contributions, or engage in discussions about LLM
 
 ```
 @misc{distilroberta-base-rejection-v1,
-  author = {Laiyer.ai},
+  author = {ProtectAI.com},
   title = {Fine-Tuned DistilRoberta-Base for Rejection in the output Detection},
   year = {2024},
   publisher = {HuggingFace},
-  url = {https://huggingface.co/laiyer/distilroberta-base-rejection-v1},
+  url = {https://huggingface.co/ProtectAI/distilroberta-base-rejection-v1},
 }
 ```