diff --git a/app/service/generate_image/service.py b/app/service/generate_image/service.py index 51d1c4d..79c7eb8 100644 --- a/app/service/generate_image/service.py +++ b/app/service/generate_image/service.py @@ -111,7 +111,7 @@ class GenerateImage: self.generate_data['image_url'] = str(image_url) self.redis_client.set(self.tasks_id, json.dumps(self.generate_data)) else: # 有污点 保存图片到本地 测试用 - cv2.imwrite(f"{self.tasks_id}.png", image_result) + cv2.imwrite(f"generate_result/{self.tasks_id}.png", image_result) self.generate_data['status'] = "SUCCESS" self.generate_data['message'] = "success" self.generate_data['image_url'] = str(GI_SYS_IMAGE_URL)