* add fmops model

README.md

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

app.py

@@ -60,6 +60,9 @@ deepset_classifier = init_prompt_injection_model(
     "laiyer/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")
+fmops_classifier = init_prompt_injection_model(
+    "laiyer/fmops-distilbert-prompt-injection-onnx"
+)  # ONNX version of fmops/distilbert-prompt-injection
 
 
 def detect_hf(prompt: str, threshold: float = 0.5, classifier=laiyer_classifier) -> (bool, bool):
@@ -88,6 +91,10 @@ def detect_hf_deepset(prompt: str) -> (bool, bool):
     return detect_hf(prompt, classifier=deepset_classifier)
 
 
+def detect_hf_fmops(prompt: str) -> (bool, bool):
+    return detect_hf(prompt, classifier=fmops_classifier)
+
+
 def detect_lakera(prompt: str) -> (bool, bool):
     try:
         response = requests.post(
@@ -130,6 +137,7 @@ def detect_rebuff(prompt: str) -> (bool, bool):
 detection_providers = {
     "Laiyer (HF model)": detect_hf_laiyer,
     "Deepset (HF model)": detect_hf_deepset,
+    "FMOps (HF model)": detect_hf_fmops,
     "Lakera Guard": detect_lakera,
     "Automorphic Aegis": detect_automorphic,
     "Rebuff": detect_rebuff,