control layout

app.py

@@ -14,16 +14,17 @@ with gr.Blocks() as block:
 
         with gr.Tab("Single Analogical Reasoning"):
             with gr.Row():
-                head_image = gr.Image(type='pil', label="Head Image")
-                head_ent = gr.Textbox(lines=1, label="Head Entity")
+                with gr.Column():
+                    head_image = gr.Image(type='pil', label="Head Image")
+                    head_ent = gr.Textbox(lines=1, label="Head Entity")
                 with gr.Column():
                     tail_image = gr.Image(type='pil', label="Tail Image")
                     tail_ent = gr.Textbox(lines=1, label="Tail Entity")
-                    with gr.Column():
-                        question_text = gr.Textbox(lines=1, label="Question Entity")
-                        question_ent = gr.Textbox(lines=1, label="Question Entity")
+                with gr.Column():
+                    question_text = gr.Textbox(lines=1, label="Question Name")
+                    question_ent = gr.Textbox(lines=1, label="Question Entity")
             submit_btn = gr.Button("Submit")
-            output_text = gr.Textbox(label="Answer Name")
+            output_text = gr.Textbox(label="Output")
 
             # examples=[['example01.jpg', MODELS[0], 'best'], ['example02.jpg', MODELS[0], 'best']]
             # ex = gr.Examples(