Update app.py

app.py

@@ -346,20 +346,21 @@ with gr.Blocks(css="style.css") as demo:
             outputs=seg_outputs,
         )
     with gr.Tab("Text Generation"):
-        with gr.Column():
-            image = gr.Image(type="pil")
-            text_input = gr.Text(label="Input Text")
-
-            text_output = gr.Text(label="Text Output")
-            chat_btn = gr.Button()
-            tokens = gr.Slider(
-                label="Max New Tokens",
-                info="Set to larger for longer generation.",
-                minimum=10,
-                maximum=100,
-                value=50,
-                step=10,
-            )
+        with gr.Row():
+            with gr.Column():
+                image = gr.Image(type="pil")
+            with gr.Column()    
+                text_input = gr.Text(label="Input Text")
+                text_output = gr.Text(label="Text Output")
+                tokens = gr.Slider(
+                    label="Max New Tokens",
+                    info="Set to larger for longer generation.",
+                    minimum=10,
+                    maximum=100,
+                    value=50,
+                    step=10,
+                )
+        chat_btn = gr.Button()
 
         chat_inputs = [
             image,