Fix scanning

zixaphir · Mar 27, 2024 · 924a6bb · 924a6bb
1 parent 175b1f6
commit 924a6bb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/ch_lib/civitai.py b/scripts/ch_lib/civitai.py
@@ -381,7 +381,7 @@ def verify_preview(path, img_dict, max_size_preview, nsfw_preview_threshold):
     image_rating = img_dict.get("nsfwLevel", 32)
     if image_rating > 1:
         util.printD(f"This image is NSFW: {image_rating}")
-        if civitai.NSFW_LEVELS[nsfw_preview_threshold] < image_rating:
+        if NSFW_LEVELS[nsfw_preview_threshold] < image_rating:
             util.printD("Skip NSFW image")
             yield (False, None)