OwenElliott commited on
Commit
40a6979
·
verified ·
1 Parent(s): a52a60c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -0
README.md CHANGED
@@ -41,6 +41,11 @@ print("Probabilities:", output[0])
41
  print("Class:", class_names[output[0].argmax()])
42
  ```
43
 
 
 
 
 
 
44
 
45
  ## Training Details
46
 
 
41
  print("Class:", class_names[output[0].argmax()])
42
  ```
43
 
44
+ ## Evaluation
45
+
46
+ This model outperforms existing NSFW detectors on our dataset, here we provide an evaluation against [AdamCodd/vit-base-nsfw-detector](https://huggingface.co/AdamCodd/vit-base-nsfw-detector) and [Falconsai/nsfw_image_detection](https://huggingface.co/Falconsai/nsfw_image_detection):
47
+
48
+ ![Evaluation against other models](./images/Comparison.png)
49
 
50
  ## Training Details
51