is-it-nsfw_ai-moderator

from transformers import AutoImageProcessor, AutoModelForImageClassification
import torch
from PIL import Image

model_id = "najicreator90856/is-it-nsfw_ai-moderator"
processor = AutoImageProcessor.from_pretrained(model_id)
model = AutoModelForImageClassification.from_pretrained(model_id)

def predict(image_path):
    image = Image.open(image_path).convert("RGB")
    inputs = processor(images=image, return_tensors="pt")
    with torch.no_grad():
        logits = model(**inputs).logits
    
    predicted_label = logits.argmax(-1).item()
    return model.config.id2label[predicted_label]

# print(predict("your_image.jpg"))