From 0083b38dcc53f073aeaf998349849dac73038b54 Mon Sep 17 00:00:00 2001 From: bennykok Date: Tue, 17 Sep 2024 20:44:44 -0700 Subject: [PATCH] chore: log image size --- custom_routes.py | 1 + 1 file changed, 1 insertion(+) diff --git a/custom_routes.py b/custom_routes.py index 4d7414e..7c98e41 100644 --- a/custom_routes.py +++ b/custom_routes.py @@ -1240,6 +1240,7 @@ async def upload_file(prompt_id, filename, subfolder=None, content_type="image/p target_url = f"{file_upload_endpoint}?file_name={filename}&run_id={prompt_id}&type={content_type}&version=v2" start_time = time.time() # Start timing here + logger.info(f"Image size: {size}") logger.info(f"Target URL: {target_url}") result = await async_request_with_retry("GET", target_url, disable_timeout=True, token=token) end_time = time.time() # End timing after the request is complete