Merge pull request #1 from wseagar/ws/nsfw_fix

fix: is_nsfw_img is None rather than false
2 years ago · 8c064587d9
parent a1675fa599 56fc414724
commit 8c064587d9
1 changed files with 1 additions and 2 deletions
--- 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))