Better blocks formatting

app.py

@@ -124,16 +124,19 @@ with gr.Blocks() as demo:
     gr.Markdown(title)
     gr.Markdown(description)
 
-    # Get the input file and slice slider as inputs
-    input_file = gr.File(label='input file')
-    z_axis = gr.Slider(0, 200, label='z-axis', value=50)
+    with gr.Row():
+        with gr.Column():
+            # Get the input file and slice slider as inputs
+            input_file = gr.File(label='input file')
+            z_axis = gr.Slider(0, 200, label='z-axis', value=50)
+            
+            # Show the button with custom label
+            button = gr.Button("Segment Tumor!")
     
-    # Show the button with custom label
-    button = gr.Button("Segment Tumor!")
-    
-    # Show the input image with different MR sequences
-    input_image = gr.Gallery(label='input MRI sequences (T1+, T1, T2, FLAIR)')
-    output_segmentation = gr.Gallery(label='output segmentations (TC, EC, WT)')
+        with gr.Column():
+            # Show the input image with different MR sequences
+            input_image = gr.Gallery(label='input MRI sequences (T1+, T1, T2, FLAIR)')
+            output_segmentation = gr.Gallery(label='output segmentations (TC, EC, WT)')
     
     # Run prediction on button click
     button.click(