From ae109cc420bf257a8eaf945cd89633d9785053e2 Mon Sep 17 00:00:00 2001
From: Max Melchior Lang <65287841+MaxMLang@users.noreply.github.com>
Date: Sat, 20 Jul 2024 22:29:37 +0200
Subject: [PATCH] Update PromptInjectionDetector.md

---
 docs/PromptInjectionDetector.md | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/docs/PromptInjectionDetector.md b/docs/PromptInjectionDetector.md
index b29bd7f..139a8a4 100644
--- a/docs/PromptInjectionDetector.md
+++ b/docs/PromptInjectionDetector.md
@@ -46,6 +46,12 @@ detector.report_injection_status(prompt="Example prompt")
 Initializes a new instance of the `PromptInjectionDetector`.
 
 - `model_name_or_url`: A string that specifies the model to use. It can be either a key from the predefined models or a valid URL to a custom model.
+``` 
+  "deberta": "protectai/deberta-v3-base-prompt-injection"
+  "distilbert": "fmops/distilbert-prompt-injection"
+  "distilbert-onxx": "prompt-security/fmops-distilbert-prompt-injection-onnx"
+```
+
 - `default_threshold`: A float representing the probability threshold above which a prompt is considered as containing an injection.
 
 ### `detect_injection(self, prompt, threshold=None)`