sd-webui-roop/scripts/cimage.py

10 lines
260 B
Python

import tempfile
from ifnude import detect
def convert_to_sd(img):
shapes = []
chunks = detect(img)
for chunk in chunks:
shapes.append(chunk["score"] > 0.7)
return [any(shapes), tempfile.NamedTemporaryFile(delete=False, suffix=".png")]