layout for 3d and -depth

app.py

@@ -180,7 +180,6 @@ def create_vis_demo():
 def predict_images_3d(dict, depth, prompt="", negative_prompt="", guidance_scale=7.5, steps=20, strength=1.0, scheduler="EulerDiscreteScheduler", keep_edges=False):
     if negative_prompt == "":
         negative_prompt = None
-    scheduler_class_name = scheduler.split("-")[0]
 
     init_image = cv2.resize(dict["image"], (512, 512))
 
@@ -209,11 +208,14 @@ def predict_images_3d(dict, depth, prompt="", negative_prompt="", guidance_scale
 
     #init_image
     #depth_image
-    output_depth = Image.fromarray(output_depth_vis)
     output_image = output.rgb[0]
 
-    output_mesh = get_mesh(output_depth_vis, output_image, keep_edges=keep_edges)
-    input_mesh = get_mesh(np.array(depth_image),init_image, keep_edges=keep_edges)
+    mesh_depth = output_depth_vis.max() - output_depth_vis
+    output_mesh = get_mesh(mesh_depth, output_image, keep_edges=keep_edges)
+
+    depth_image = np.array(depth_image)
+    depth_image_mesh = depth_image.max() - depth_image
+    input_mesh = get_mesh(depth_image_mesh,init_image, keep_edges=keep_edges)
     
     return input_mesh, output_mesh, gr.update(visible=True)
 
@@ -221,15 +223,12 @@ def create_3d_demo():
 
     gr.Markdown("### Image to 3D mesh")
 
-    with gr.Column():
+    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)
             checkbox = gr.Checkbox(label="Keep occlusion edges", value=False)
-
-            with gr.Row(elem_id="prompt-container", mobile_collapse=False, equal_height=True):
-                with gr.Row():
-                    prompt = gr.Textbox(placeholder="Your prompt (what you want in place of what is erased)", show_label=False, elem_id="prompt")
-                    btn = gr.Button("Inpaint!", elem_id="run_button")
+            prompt = gr.Textbox(placeholder="Your prompt (what you want in place of what is erased)", show_label=False, elem_id="prompt")
             
             with gr.Accordion(label="Advanced Settings", open=False):
                 with gr.Row(mobile_collapse=False, equal_height=True):
@@ -241,8 +240,8 @@ def create_3d_demo():
                     schedulers = ["DEISMultistepScheduler", "HeunDiscreteScheduler", "EulerDiscreteScheduler", "DPMSolverMultistepScheduler", "DPMSolverMultistepScheduler-Karras", "DPMSolverMultistepScheduler-Karras-SDE"]
                     scheduler = gr.Dropdown(label="Schedulers", choices=schedulers, value="EulerDiscreteScheduler")
     
-    with gr.Column() as share_btn_container:
-        with gr.Row():
+    with gr.Row() as share_btn_container:
+        with gr.Column():
             result_og = gr.Model3D(label="original 3d reconstruction", clear_color=[
                                                     1.0, 1.0, 1.0, 1.0])