* move models

app.py

@@ -59,16 +59,16 @@ def convert_elapsed_time(diff_time) -> float:
 
 
 deepset_classifier = init_prompt_injection_model(
-    "laiyer/deberta-v3-base-injection-onnx"
+    "ProtectAI/deberta-v3-base-injection-onnx"
 )  # ONNX version of deepset/deberta-v3-base-injection
-laiyer_classifier = init_prompt_injection_model("laiyer/deberta-v3-base-prompt-injection", "onnx")
+protectai_classifier = init_prompt_injection_model("ProtectAI/deberta-v3-base-prompt-injection", "onnx")
 fmops_classifier = init_prompt_injection_model(
-    "laiyer/fmops-distilbert-prompt-injection-onnx"
+    "ProtectAI/fmops-distilbert-prompt-injection-onnx"
 )  # ONNX version of fmops/distilbert-prompt-injection
 
 
 def detect_hf(
-    prompt: str, threshold: float = 0.5, classifier=laiyer_classifier, label: str = "INJECTION"
+    prompt: str, threshold: float = 0.5, classifier=protectai_classifier, label: str = "INJECTION"
 ) -> (bool, bool):
     try:
         pi_result = classifier(prompt)
@@ -85,8 +85,8 @@ def detect_hf(
         return False, False
 
 
-def detect_hf_laiyer(prompt: str) -> (bool, bool):
-    return detect_hf(prompt, classifier=laiyer_classifier)
+def detect_hf_protectai(prompt: str) -> (bool, bool):
+    return detect_hf(prompt, classifier=protectai_classifier)
 
 
 def detect_hf_deepset(prompt: str) -> (bool, bool):
@@ -156,7 +156,7 @@ def detect_azure(prompt: str) -> (bool, bool):
 
 
 detection_providers = {
-    "Laiyer (HF model)": detect_hf_laiyer,
+    "ProtectAI (HF model)": detect_hf_protectai,
     "Deepset (HF model)": detect_hf_deepset,
     "FMOps (HF model)": detect_hf_fmops,
     "Lakera Guard": detect_lakera,
@@ -196,7 +196,7 @@ def execute(prompt: str) -> List[Union[str, bool, float]]:
     hf_api.upload_file(
         path_or_fileobj=fileobj,
         path_in_repo=result_path,
-        repo_id="laiyer/prompt-injection-benchmark",
+        repo_id="ProtectAI/prompt-injection-benchmark",
         repo_type="dataset",
     )
     logger.info(f"Stored prompt: {prompt}")
@@ -236,7 +236,7 @@ if __name__ == "__main__":
         "<br /><br />"
         "HuggingFace (HF) models are hosted on Spaces while other providers are called as APIs.<br /><br />"
         "<a href=\"https://join.slack.com/t/laiyerai/shared_invite/zt-28jv3ci39-sVxXrLs3rQdaN3mIl9IT~w\">Join our Slack community to discuss LLM Security</a><br />"
-        "<a href=\"https://github.com/laiyer-ai/llm-guard\">Secure your LLM interactions with LLM Guard</a>",
+        "<a href=\"https://github.com/protectai/llm-guard\">Secure your LLM interactions with LLM Guard</a>",
         examples=[
             [
                 example,