From 7970e27901515a7e33ce37a96baa1737b52980da Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Thireus=20=E2=98=A0?= Date: Sun, 2 Apr 2023 20:03:59 +0100 Subject: [PATCH] Fixes #10 --- scripts/postprocessing_rembg.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/postprocessing_rembg.py b/scripts/postprocessing_rembg.py index da80e86..50b3fec 100644 --- a/scripts/postprocessing_rembg.py +++ b/scripts/postprocessing_rembg.py @@ -45,7 +45,7 @@ class ScriptPostprocessingUpscale(scripts_postprocessing.ScriptPostprocessing): } def process(self, pp: scripts_postprocessing.PostprocessedImage, model, return_mask, alpha_matting, alpha_matting_foreground_threshold, alpha_matting_background_threshold, alpha_matting_erode_size): - if model == "None": + if not model or model == "None": return pp.image = rembg.remove(