From 56fc41472448421400c44ddc2839a5229039dea6 Mon Sep 17 00:00:00 2001 From: wseagar Date: Thu, 15 Sep 2022 20:11:36 +1200 Subject: [PATCH] fix: is_nsfw_img is None rather than false --- imaginairy/api.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/imaginairy/api.py b/imaginairy/api.py index 0aaafa8..e94eb30 100755 --- a/imaginairy/api.py +++ b/imaginairy/api.py @@ -253,8 +253,7 @@ def imagine( upscaled_img = None is_nsfw_img = None if IMAGINAIRY_SAFETY_MODE != SafetyMode.DISABLED: - if is_nsfw(img, x_sample, half_mode=half_mode): - is_nsfw_img = True + is_nsfw_img = is_nsfw(img, x_sample, half_mode=half_mode) if is_nsfw_img and IMAGINAIRY_SAFETY_MODE == SafetyMode.FILTER: logger.info(" ⚠️ Filtering NSFW image") img = img.filter(ImageFilter.GaussianBlur(radius=40))