Update README.md
Browse files
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 |
+

|
49 |
|
50 |
## Training Details
|
51 |
|