Fix: 필기 제거 기능 사용성 개선

AI-SIP · Nov 24, 2024 · 6c07124 · 6c07124
1 parent e3a400c
commit 6c07124
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/app/AIProcessor.py b/app/AIProcessor.py
@@ -187,6 +187,7 @@ def process(self, img_bytes, user_inputs, user_labels, extension='jpg'):
         else:
             logging.info("***** 사용자 입력 세그멘테이션 스킵 ******")
             masks_by_user = None
+            mask_bytes = None
         # cv2.imwrite('test_images/mask_total.png', masks_total)
         if isinstance(masks_total, np.ndarray):
             image_output = self.inpainting(image_output, masks_total)