diff --git a/custom_routes.py b/custom_routes.py index 6575ebf..047c5bc 100644 --- a/custom_routes.py +++ b/custom_routes.py @@ -1365,6 +1365,8 @@ async def upload_file(prompt_id, filename, subfolder=None, content_type="image/p item["upload_duration"] = end_time - start_time if ok.get("is_public") is not None: item["is_public"] = ok.get("is_public") + + return item def have_pending_upload(prompt_id): if prompt_id in prompt_metadata and len(prompt_metadata[prompt_id].uploading_nodes) > 0: