generate 模型名称更新

This commit is contained in:
zchen
2024-06-23 16:30:18 +08:00
parent 9adabc3f36
commit 8fb75b0390

View File

@@ -103,7 +103,7 @@ class GenerateImage:
image_result = not_smudge_image image_result = not_smudge_image
if is_smudge: # 无污点 if is_smudge: # 无污点
# image_result = adjust_contrast(image_result) # 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}") # logger.info(f"upload image SUCCESS {image_url}")
self.generate_data['status'] = "SUCCESS" self.generate_data['status'] = "SUCCESS"
self.generate_data['message'] = "success" self.generate_data['message'] = "success"