EridAI-SensitiveFactor

Модель распределяющая изображения на Безопасный и Чувствительный (NSFW) факторы.

Использование

from PIL import Image
model = torch.load("pytorch_model.bin")
img = Image.open("photo.jpg")
pred = model(img)  # 0=Clear, 1=Sensitive