Reformat block layout

app.py

@@ -22,7 +22,7 @@ title = '<h1 style="text-align: center;">Segment Brain Tumors with MONAI! 🧠 <
 description = """
 ## To run 🚀
 
-Upload a image file in the format: 4 channel MRI (4 aligned MRIs T1c, T1, T2, FLAIR at 1x1x1 mm), or try out one of the examples below! 
+Upload a brain MRI image file, or try out one of the examples below! 
 If you want to see a different slice, update the slider and click the button.
 
 More details on the model can be found [here!](https://huggingface.co/katielink/brats_mri_segmentation_v0.1.0)
@@ -41,7 +41,7 @@ references = """
 """
 
 examples = [
-    ['examples/BRATS_485.nii.gz', 100],
+    ['examples/BRATS_485.nii.gz', 65],
     ['examples/BRATS_486.nii.gz', 100]
 ]
 
@@ -125,18 +125,23 @@ with gr.Blocks() as demo:
     gr.Markdown(description)
 
     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)
-    
+        # 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():
+        # Show the button with custom label
+        button = gr.Button("Segment Tumor!")
+
+    with gr.Row():
         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)')
-    
-    # Show the button with custom label
-    button = gr.Button("Segment Tumor!")
+        
+        with gr.Column():
+            # Show the segmentation labels
+            output_segmentation = gr.Gallery(label='output segmentations (TC, WT, ET)')
+
     
     # Run prediction on button click
     button.click(