* introduce new model

README.md

@@ -4,7 +4,7 @@ emoji: 📝
 colorFrom: yellow
 colorTo: gray
 sdk: gradio
-sdk_version: 4.19.1
+sdk_version: 4.27.0
 pinned: true
 license: apache-2.0
 ---

app.py

@@ -64,16 +64,19 @@ def convert_elapsed_time(diff_time) -> float:
 deepset_classifier = init_prompt_injection_model(
     "ProtectAI/deberta-v3-base-injection-onnx"
 )  # ONNX version of deepset/deberta-v3-base-injection
-protectai_classifier = init_prompt_injection_model(
+protectai_v1_classifier = init_prompt_injection_model(
     "ProtectAI/deberta-v3-base-prompt-injection", "onnx"
 )
+protectai_v2_classifier = init_prompt_injection_model(
+    "ProtectAI/deberta-v3-base-prompt-injection-v2", "onnx"
+)
 fmops_classifier = init_prompt_injection_model(
     "ProtectAI/fmops-distilbert-prompt-injection-onnx"
 )  # ONNX version of fmops/distilbert-prompt-injection
 
 
 def detect_hf(
-    prompt: str, threshold: float = 0.5, classifier=protectai_classifier, label: str = "INJECTION"
+    prompt: str, threshold: float = 0.5, classifier=protectai_v1_classifier, label: str = "INJECTION"
 ) -> (bool, bool):
     try:
         pi_result = classifier(prompt)
@@ -90,8 +93,12 @@ def detect_hf(
         return False, False
 
 
-def detect_hf_protectai(prompt: str) -> (bool, bool):
-    return detect_hf(prompt, classifier=protectai_classifier)
+def detect_hf_protectai_v1(prompt: str) -> (bool, bool):
+    return detect_hf(prompt, classifier=protectai_v1_classifier)
+
+
+def detect_hf_protectai_v2(prompt: str) -> (bool, bool):
+    return detect_hf(prompt, classifier=protectai_v2_classifier)
 
 
 def detect_hf_deepset(prompt: str) -> (bool, bool):
@@ -191,14 +198,15 @@ def detect_aws_comprehend(prompt: str) -> (bool, bool):
 
 
 detection_providers = {
-    "ProtectAI (HF model)": detect_hf_protectai,
+    "ProtectAI v1 (HF model)": detect_hf_protectai_v1,
+    "ProtectAI v2 (HF model)": detect_hf_protectai_v2,
     "Deepset (HF model)": detect_hf_deepset,
     "FMOps (HF model)": detect_hf_fmops,
     "Lakera Guard": detect_lakera,
     "Automorphic Aegis": detect_automorphic,
     # "Rebuff": detect_rebuff,
     "Azure Content Safety": detect_azure,
-    "AWS Comprehend": detect_aws_comprehend,
+    #"AWS Comprehend": detect_aws_comprehend,
 }
 
 

requirements.txt

@@ -1,9 +1,9 @@
-boto3==1.34.44
+boto3==1.34.88
 git+https://github.com/automorphic-ai/aegis.git
-gradio==4.19.1
-huggingface_hub==0.20.3
-onnxruntime==1.17.0
-optimum[onnxruntime]==1.17.1
+gradio==4.27.0
+huggingface_hub==0.22.2
+onnxruntime==1.17.3
+optimum[onnxruntime]==1.19.0
 rebuff==0.1.1
 requests==2.31.0
-transformers==4.37.2
+transformers==4.40.0