#!/usr/bin/env python3
import PIL
from transformers import CLIPImageProcessor
from diffusers.pipelines.stable_diffusion import StableDiffusionSafetyChecker

feature_extractor = CLIPImageProcessor.from_pretrained("runwayml/stable-diffusion-v1-5", subfolder="feature_extractor")
safety_checker = StableDiffusionSafetyChecker.from_pretrained("runwayml/stable-diffusion-v1-5", subfolder="safety_checker")
device = "cpu"

image = PIL.Image.open("/home/patrick/images/0.png")

safety_checker_input = feature_extractor(image, return_tensors="pt").to(device)
image, has_nsfw_concept = safety_checker(images=image, clip_input=safety_checker_input.pixel_values)