Update README.md
Browse files
README.md
CHANGED
|
@@ -48,6 +48,7 @@ print(classifier("this is a prompt-injection prompt"))
|
|
| 48 |
The dataset used for training the model was taken from
|
| 49 |
|
| 50 |
[wildguardmix/train](https://huggingface.co/datasets/allenai/wildguardmix)
|
|
|
|
| 51 |
[safe-guard-prompt-injection/train](https://huggingface.co/datasets/xTRam1/safe-guard-prompt-injection)
|
| 52 |
|
| 53 |
### Training Procedure
|
|
@@ -89,6 +90,7 @@ Supervised finetuning with above dataset
|
|
| 89 |
The dataset used for training the model was taken from
|
| 90 |
|
| 91 |
[wildguardmix/test](https://huggingface.co/datasets/allenai/wildguardmix)
|
|
|
|
| 92 |
[safe-guard-prompt-injection/test](https://huggingface.co/datasets/xTRam1/safe-guard-prompt-injection)
|
| 93 |
|
| 94 |
### Results
|
|
|
|
| 48 |
The dataset used for training the model was taken from
|
| 49 |
|
| 50 |
[wildguardmix/train](https://huggingface.co/datasets/allenai/wildguardmix)
|
| 51 |
+
and
|
| 52 |
[safe-guard-prompt-injection/train](https://huggingface.co/datasets/xTRam1/safe-guard-prompt-injection)
|
| 53 |
|
| 54 |
### Training Procedure
|
|
|
|
| 90 |
The dataset used for training the model was taken from
|
| 91 |
|
| 92 |
[wildguardmix/test](https://huggingface.co/datasets/allenai/wildguardmix)
|
| 93 |
+
and
|
| 94 |
[safe-guard-prompt-injection/test](https://huggingface.co/datasets/xTRam1/safe-guard-prompt-injection)
|
| 95 |
|
| 96 |
### Results
|