AdamCodd commited on
Commit
5ae75b7
1 Parent(s): 6abea56

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -49,7 +49,9 @@ It achieves the following results on the evaluation set:
49
 
50
  ## Model description
51
 
52
- This model was made to detect NSFW Stable-diffusion prompts by training it on 2M SFW and NSFW prompts (1043475 samples of each, perfectly balanced). One words prompts have been excluded to remove useless noise for the prediction.
 
 
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