Update app.py

app.py

@@ -223,11 +223,11 @@ def predict_images_3d(dict, depth, prompt="", negative_prompt="", guidance_scale
     output = pipe(prompt = prompt, negative_prompt=negative_prompt, image=init_image, mask_image=mask, depth_image=depth_image, guidance_scale=guidance_scale, num_inference_steps=int(steps), strength=strength)
     
     # resize to original size
-    depth_image = depth_image.resize(og_size)
-    output_depth = output.depth[0].resize(og_size)
+    #depth_image = depth_image.resize(og_size)
+    #output_depth = output.depth[0].resize(og_size)
     
     depth_in = denormalize(np.array(depth_image), min, max)
-    depth_out = denormalize(np.array(output_depth), min, max)
+    depth_out = denormalize(np.array(output.depth[0]), min, max)
     
     output_image = output.rgb[0]
 
@@ -243,8 +243,8 @@ def create_3d_demo():
     with gr.Row():
         with gr.Row():
             with gr.Column():
-                image = gr.Image(source='upload', tool='sketch', elem_id="image_upload", type="numpy", label="Upload",height=400)
-                depth = gr.Image(source='upload', elem_id="depth_upload", type="numpy", label="Upload",height=400)
+                image = gr.Image(source='upload', tool='sketch', elem_id="image_upload", type="numpy", label="Upload",height=400, shape=(512,512))
+                depth = gr.Image(source='upload', elem_id="depth_upload", type="numpy", label="Upload",height=400, shape=(512,512))
                 checkbox = gr.Checkbox(label="Keep occlusion edges", value=False)
                 prompt = gr.Textbox(placeholder="Your prompt (what you want in place of what is erased)", show_label=False, elem_id="prompt")