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

2024-04-23 15:41:59 +08:00
parent 8b5cf60d07
commit 51540d2970
1 changed files with 1 additions and 0 deletions
--- a/app/service/generate_image/service.py
+++ b/app/service/generate_image/service.py
@@ -137,6 +137,7 @@ class GenerateImage:
                    break
                time_out -= 1
                time.sleep(0.1)
+                logger.info(time_out, generate_data)
            return generate_data
        except Exception as e:
            self.generate_data['status'] = "FAILURE"