bugfix

app.py

@@ -103,7 +103,6 @@ def upload_image_to_s3(image, account_id, access_key, secret_key, bucket_name):
     buffer = BytesIO()
     image.save(buffer, "PNG")
     buffer.seek(0)
-    print("start to upload")
     s3.upload_fileobj(buffer, bucket_name, image_file)
     print("upload finish", image_file)
     return image_file
@@ -112,16 +111,18 @@ 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):
-
+    print("process start")
     if image_url:
+        print(image_url)
         orginal_image = load_image(image_url)
     else:
         orginal_image = Image.fromarray(image)
 
     size = (orginal_image.size[0], orginal_image.size[1])
-    print(size)
+    print("image size", size)
     depth_image = get_depth_map(orginal_image)
     generator = torch.Generator().manual_seed(seed)
+    print(prompt, n_prompt, guidance_scale, num_steps, control_strength, )
     generated_image = pipe(
         prompt=prompt,
         negative_prompt=n_prompt,
@@ -140,7 +141,7 @@ def process(image, image_url, prompt, n_prompt, num_steps, guidance_scale, contr
     else:
         result = {"status": "success", "message": "Image generated but not uploaded"}
     
-    return [[depth_image, generated_image], json.dumps(result)]
+    return [[orginal_image, generated_image], json.dumps(result)]
 
 with gr.Blocks() as demo:
     
@@ -171,7 +172,7 @@ with gr.Blocks() as demo:
         
 
         with gr.Column():
-            result = ImageSlider(label="Generate image", type="pil", slider_color="pink")
+            images = ImageSlider(label="Generate images", type="pil", slider_color="pink")
             logs = gr.Textbox(label="logs")
             
     inputs = [
@@ -198,7 +199,7 @@ with gr.Blocks() as demo:
         ).then(
             fn=process,
             inputs=inputs,
-            outputs=[result, logs],
+            outputs=[images, logs],
             api_name=False
         )