10 lines
260 B
Python
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")]
|