From d090f9d0726b103c39e983e8e4530d904dddddcf Mon Sep 17 00:00:00 2001 From: Bryce Drennan Date: Fri, 23 Sep 2022 15:15:41 -0700 Subject: [PATCH] fix: unneeded argument --- imaginairy/api.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/imaginairy/api.py b/imaginairy/api.py index fdbc8a5..bdbdf39 100755 --- a/imaginairy/api.py +++ b/imaginairy/api.py @@ -310,7 +310,7 @@ def imagine( caption = generate_caption(img) logger.info(f" Generated caption: {caption}") if IMAGINAIRY_SAFETY_MODE != SafetyMode.DISABLED: - is_nsfw_img = is_nsfw(img, x_sample) + is_nsfw_img = is_nsfw(img) if is_nsfw_img and IMAGINAIRY_SAFETY_MODE == SafetyMode.FILTER: logger.info(" ⚠️ Filtering NSFW image") img = img.filter(ImageFilter.GaussianBlur(radius=40))