feat generate product image 部署

2024-06-04 15:33:34 +08:00
parent 40a2e158e2
commit 1d94d485e9
5 changed files with 108 additions and 114 deletions
--- a/app/service/generate_image/utils/upload_sd_image.py
+++ b/app/service/generate_image/utils/upload_sd_image.py
@@ -34,7 +34,7 @@ s3 = boto3.client('s3', aws_access_key_id=S3_ACCESS_KEY, aws_secret_access_key=S
 #     except Exception as e:
 #         print(f'上传到 S3 失败: {e}')

-def upload_single_logo(image, user_id, category, object_name):
+def upload_SDXL_image(image, user_id, category, object_name):
    try:
        image_data = io.BytesIO()
        image.save(image_data, format='PNG')