diff --git a/scripts/ch_lib/civitai.py b/scripts/ch_lib/civitai.py index 2a15ae7..0efd71b 100644 --- a/scripts/ch_lib/civitai.py +++ b/scripts/ch_lib/civitai.py @@ -342,7 +342,7 @@ def get_preview_image_by_model_path(model_path:str, max_size_preview, skip_nsfw_ if model_info["images"]: for img_dict in model_info["images"]: if "nsfw" in img_dict.keys(): - if img_dict["nsfw"]: + if img_dict["nsfw"] and img_dict["nsfw"] != "None": util.printD("This image is NSFW") if skip_nsfw_preview: util.printD("Skip NSFW image")