bugfix

app.py

@@ -66,7 +66,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     return seed
 
 
-def get_depth_map(image):
+def get_depth_map(image, progress):
     original_size = (image.size[1], image.size[0]) 
     image = feature_extractor(images=image, return_tensors="pt").pixel_values.to("cuda")
     with torch.no_grad(), torch.autocast("cuda"):
@@ -86,7 +86,7 @@ def get_depth_map(image):
     return image
 
 
-def upload_image_to_s3(image, account_id, access_key, secret_key, bucket_name):
+def upload_image_to_s3(image, account_id, access_key, secret_key, bucket_name, progress):
     print("upload_image_to_s3", account_id, access_key, secret_key, bucket_name)
     connectionUrl = f"https://{account_id}.r2.cloudflarestorage.com"
 
@@ -110,7 +110,7 @@ def upload_image_to_s3(image, account_id, access_key, secret_key, bucket_name):
 
 
 @spaces.GPU(enable_queue=True)
-def process(image, image_url, prompt, n_prompt, num_steps, guidance_scale, control_strength, seed, upload_to_s3, account_id, access_key, secret_key, bucket):
+def process(image, image_url, prompt, n_prompt, num_steps, guidance_scale, control_strength, seed, upload_to_s3, account_id, access_key, secret_key, bucket, progress=gr.Progress(track_tqdm=True)):
     print("process start")
     if image_url:
         print(image_url)
@@ -120,9 +120,9 @@ def process(image, image_url, prompt, n_prompt, num_steps, guidance_scale, contr
 
     size = (orginal_image.size[0], orginal_image.size[1])
     print("image size", size)
-    depth_image = get_depth_map(orginal_image)
+    depth_image = get_depth_map(orginal_image, progress)
     generator = torch.Generator().manual_seed(seed)
-    print(prompt, n_prompt, guidance_scale, num_steps, control_strength, )
+    print(prompt, n_prompt, guidance_scale, num_steps, control_strength)
     generated_image = pipe(
         prompt=prompt,
         negative_prompt=n_prompt,
@@ -136,12 +136,12 @@ def process(image, image_url, prompt, n_prompt, num_steps, guidance_scale, contr
     ).images[0]
 
     if upload_to_s3:
-        url = upload_image_to_s3(generated_image, account_id, access_key, secret_key, bucket)
+        url = upload_image_to_s3(generated_image, account_id, access_key, secret_key, bucket, progress)
         result = {"status": "success", "url": url}
     else:
         result = {"status": "success", "message": "Image generated but not uploaded"}
     
-    return [[orginal_image, generated_image], json.dumps(result)]
+    return [orginal_image, generated_image], json.dumps(result)
 
 with gr.Blocks() as demo: