keep safe images
This commit is contained in:
parent
2eb90ba559
commit
99bd6b0708
|
@ -58,10 +58,14 @@ class EditSafetyStage(BaseStage):
|
|||
if check.is_csam:
|
||||
logger.warning("flagging csam result: %s, %s", i, prompt)
|
||||
is_csam = True
|
||||
continue
|
||||
|
||||
if check.is_nsfw and block_nsfw:
|
||||
logger.warning("blocking nsfw image: %s, %s", i, prompt)
|
||||
results.append(Image.new("RGB", image.size, color="black"))
|
||||
continue
|
||||
|
||||
results.append(image)
|
||||
|
||||
if is_csam:
|
||||
logger.warning("blocking csam result")
|
||||
|
|
|
@ -132,10 +132,16 @@ def run_txt2img_pipeline(
|
|||
)
|
||||
|
||||
# add a thumbnail, if requested
|
||||
cover = images.as_images()[0]
|
||||
if params.thumbnail and (
|
||||
cover.width > server.thumbnail_size or cover.height > server.thumbnail_size
|
||||
result_size = images.size()
|
||||
if (
|
||||
params.thumbnail
|
||||
and len(images) > 0
|
||||
and (
|
||||
result_size.width > server.thumbnail_size
|
||||
or result_size.height > server.thumbnail_size
|
||||
)
|
||||
):
|
||||
cover = images.as_images()[0]
|
||||
thumbnail = cover.copy()
|
||||
thumbnail.thumbnail((server.thumbnail_size, server.thumbnail_size))
|
||||
|
||||
|
|
Loading…
Reference in New Issue