UI update

app.py

@@ -33,13 +33,13 @@ with block:
     # UI
     with gr.Column():
         with gr.Row():
-            input_image = gr.Image(label='Input', type='pil', tool=None)
             with gr.Column():
+                input_image = gr.Image(label='Input', type='pil', tool=None)
                 sam_encode_btn = gr.Button('Encode', variant='primary')
                 sam_encode_status = gr.Label('Not encoded yet')
             masks_annotated_image = gr.AnnotatedImage(label='Segments', height=512)
-            pcl_figure = gr.Model3D(label="3-D Reconstruction", clear_color=[1.0, 1.0, 1.0, 1.0])
             with gr.Column():
+                pcl_figure = gr.Model3D(label="3-D Reconstruction", clear_color=[1.0, 1.0, 1.0, 1.0])
                 n_samples = gr.Slider(minimum=1e3, maximum=1e6, step=1e3, default=1e3, label='Number of Samples')
                 cube_size = gr.Slider(minimum=0.000001, maximum=0.001, step=0.000001, default=0.00001, label='Cube size')
         with gr.Row():