fix layout with legend

app.py

@@ -214,19 +214,19 @@ with gr.Blocks() as demo:
             inp = gr.File()
             btn = gr.Button("Submit")
     
-    with gr.Row():
-        gr.Markdown(value='### T1')
-        gr.Markdown(value='### T2')
-        gr.Markdown(value='### T3')
-        gr.Markdown(value='### Model prediction')
-        gr.Markdown(value='### Legend')
+    # with gr.Row():
+    #     gr.Markdown(value='### T1', scale=)
+    #     gr.Markdown(value='### T2')
+    #     gr.Markdown(value='### T3')
+    #     gr.Markdown(value='### Model prediction')
+    #     gr.Markdown(value='### Legend')
         
     with gr.Row():
-        inp1=gr.Image(image_mode='RGB')
-        inp2=gr.Image(image_mode='RGB')
-        inp3=gr.Image(image_mode='RGB')
-        out = gr.Image(image_mode='L')
-        gr.Image(value='Legend.png', image_mode='RGB', scale=0.05 show_label=False, min_width=50)
+        inp1=gr.Image(image_mode='RGB', scale=10, label='T1')
+        inp2=gr.Image(image_mode='RGB', scale=10, label='T2')
+        inp3=gr.Image(image_mode='RGB', scale=10, label='T3')
+        out = gr.Image(image_mode='RGB', scale=10, label='Model prediction')
+        gr.Image(value='Legend.png', image_mode='RGB', scale=3, show_label=False)
     
     btn.click(fn=func, inputs=inp, outputs=[inp1, inp2, inp3, out])