From a403daa314dc0f8773882f0092106c07f50eb6ad Mon Sep 17 00:00:00 2001 From: bennykok Date: Tue, 17 Sep 2024 23:09:42 -0700 Subject: [PATCH] fix --- custom_routes.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/custom_routes.py b/custom_routes.py index 7f4e905..a7de81b 100644 --- a/custom_routes.py +++ b/custom_routes.py @@ -1256,22 +1256,21 @@ async def upload_file(prompt_id, filename, subfolder=None, content_type="image/p logger.info(f"Result: {ok}") async with aiofiles.open(file, 'rb') as f: - # data = await f.read() - - # size = str(len(data)) + data = await f.read() + size = str(len(data)) # logger.info(f"Image size: {size}") start_time = time.time() # Start timing here headers = { "Content-Type": content_type, - # "Content-Length": size, + "Content-Length": size, } if ok.get('include_acl') is True: headers["x-amz-acl"] = "public-read" # response = requests.put(ok.get("url"), headers=headers, data=data) - response = await async_request_with_retry('PUT', ok.get("url"), headers=headers, data=file_sender(f, chunk_size)) + response = await async_request_with_retry('PUT', ok.get("url"), headers=headers, data=data) logger.info(f"Upload file response status: {response.status}, status text: {response.reason}") end_time = time.time() # End timing after the request is complete logger.info("Upload time: {:.2f} seconds".format(end_time - start_time))