Update README.md
Browse files
README.md
CHANGED
@@ -49,7 +49,9 @@ It achieves the following results on the evaluation set:
|
|
49 |
|
50 |
## Model description
|
51 |
|
52 |
-
This model
|
|
|
|
|
53 |
|
54 |
## Usage
|
55 |
|
|
|
49 |
|
50 |
## Model description
|
51 |
|
52 |
+
This model is designed to identify NSFW prompts in Stable-diffusion, trained on a dataset comprising of ~2 million prompts, evenly split between SFW and NSFW categories (1,043,475 samples of each, ensuring a balanced dataset). Single-word prompts have been excluded to enhance the accuracy and relevance of the predictions.
|
53 |
+
|
54 |
+
Although this model demonstrates satisfactory accuracy, it is recommended to use it in conjunction with this [image NSFW detector](https://huggingface.co/AdamCodd/vit-base-nsfw-detector) to improve overall detection capabilities and minimize the occurrence of false positives.
|
55 |
|
56 |
## Usage
|
57 |
|