feat 新增生成sketch时对图片清理背景，剔除带有污点的结果图

2024-04-23 15:43:03 +08:00
parent 51540d2970
commit 2c2b0d3b0d
1 changed files with 1 additions and 0 deletions
--- a/app/service/generate_image/service.py
+++ b/app/service/generate_image/service.py
@@ -84,6 +84,7 @@ class GenerateImage:
                if is_smudge:  # 无污点
                    image_result = not_smudge_image
                    image_url = upload_png_sd(image_result, user_id=self.user_id, category=f"{self.category}", object_name=f"{self.tasks_id}.png")
+                    logger.info(f"upload image SUCCESS ： {image_url}")
                    self.generate_data['status'] = "SUCCESS"
                    self.generate_data['message'] = "success"
                    self.generate_data['data'] = str(image_url)