diff --git a/app/service/generate_image/service_generate_image.py b/app/service/generate_image/service_generate_image.py index 2ba7862..86912f8 100644 --- a/app/service/generate_image/service_generate_image.py +++ b/app/service/generate_image/service_generate_image.py @@ -91,7 +91,7 @@ class GenerateImage: image = result.as_numpy("generated_image") image_result = cv2.cvtColor(np.squeeze(image.astype(np.uint8)), cv2.COLOR_RGB2BGR) is_smudge = True - if self.category == "sketch" and self.version == "fast": + if self.category == "sketch": if self.version == "fast": # 色阶调整 cutoff = 1