Image Classification
Transformers
Safetensors
English
siglip
siglip2
Not-For-All-Audiences
art
Synthetic
nsfw
sfw
Anime Picture
Hentai
Normal
Pornography
Enticing
Sensual
Update README.md
Browse files
README.md
CHANGED
@@ -31,7 +31,7 @@ from PIL import Image
|
|
31 |
import torch
|
32 |
|
33 |
# Load model and processor
|
34 |
-
model_name = "
|
35 |
model = SiglipForImageClassification.from_pretrained(model_name)
|
36 |
processor = AutoImageProcessor.from_pretrained(model_name)
|
37 |
|
|
|
31 |
import torch
|
32 |
|
33 |
# Load model and processor
|
34 |
+
model_name = "strangerguardhf/nsfw-image-detection" # Replace with your model path if needed
|
35 |
model = SiglipForImageClassification.from_pretrained(model_name)
|
36 |
processor = AutoImageProcessor.from_pretrained(model_name)
|
37 |
|