Ateeqq
/

nsfw-image-detection

Image Classification

Not-For-All-Audiences

Model card Files Files and versions

Ateeqq commited on May 4

Commit

a6bcf3b

·

verified ·

1 Parent(s): 0394de0

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -69,7 +69,7 @@ from transformers import AutoImageProcessor, SiglipForImageClassification
 from PIL import Image
 import torch.nn.functional as F
-model_path = "Ateeqq/siglip2-safety-classifier-gpu"
 processor = AutoImageProcessor.from_pretrained(model_path)
 model = SiglipForImageClassification.from_pretrained(model_path)

 from PIL import Image
 import torch.nn.functional as F
+model_path = "Ateeqq/nsfw-image-detection"
 processor = AutoImageProcessor.from_pretrained(model_path)
 model = SiglipForImageClassification.from_pretrained(model_path)