generate 模型名称更新

2024-06-23 16:30:18 +08:00
parent 9adabc3f36
commit 8fb75b0390
1 changed files with 1 additions and 1 deletions
--- a/app/service/generate_image/service_generate_image.py
+++ b/app/service/generate_image/service_generate_image.py
@@ -103,7 +103,7 @@ class GenerateImage:
                image_result = not_smudge_image
            if is_smudge:  # 无污点
                # image_result = adjust_contrast(image_result)
-                image_url = upload_png_sd(image_result, user_id=self.user_id, category=f"{self.category}", object_name=f"{self.tasks_id}.png")
+                image_url = upload_png_sd(image_result, user_id=self.user_id, category=f"{self.category}", file_name=f"{self.tasks_id}.png")
                # logger.info(f"upload image SUCCESS ： {image_url}")
                self.generate_data['status'] = "SUCCESS"
                self.generate_data['message'] = "success"