UI update.

app.py

@@ -93,8 +93,11 @@ with gr.Blocks() as demo:
 
       with gr.Column():
 
-        gr_image_1 = gr.Image(type="numpy", source="canvas", label="Decoding")
-        gr_image_2 = gr.Image(type="numpy", source="canvas", label="Segmentation")
+        with gr.Row():
+          with gr.Column():
+            gr_image_1 = gr.Image(type="numpy", shape=(112, 112), source="canvas", label="Decoding")
+          with gr.Column():
+            gr_image_2 = gr.Image(type="numpy", shape=(112, 112), source="canvas", label="Segmentation")
 
       with gr.Column():
         gr_slot_slider = gr.Slider(1, 11, value=1, step=1, label="Slot Index",