[UPDATE] => changing the layout to singe input text that is then translated by the two selected models

app.py

@@ -51,23 +51,18 @@ def translate_fine_tuned(text, model):
 
 # Create Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("# Translation Models")
-
+    gr.Markdown("# Translation Models\nTranslate text using base and fine-tuned models.")
+    
     with gr.Row():
-        with gr.Column():
-            gr.Markdown("### Base Model")
-            base_input = gr.Textbox(placeholder="Enter text to translate", label="Input")
-            base_output = gr.Textbox(label="Translation")
-            base_translate_btn = gr.Button("Translate")
-            base_translate_btn.click(translate_base, inputs=base_input, outputs=base_output)
-
-        with gr.Column():
-            gr.Markdown("### Fine-tuned Models")
-            fine_tuned_input = gr.Textbox(placeholder="Enter text to translate", label="Input")
-            model_select = gr.Dropdown(choices=["S model", "M model", "L model"], label="Select Model")
-            fine_tuned_output = gr.Textbox(label="Translation")
-            fine_tuned_translate_btn = gr.Button("Translate")
-            fine_tuned_translate_btn.click(translate_fine_tuned, inputs=[fine_tuned_input, model_select], outputs=fine_tuned_output)
+        text_input = gr.Textbox(placeholder="Enter text to translate", label="Input", placeholder="If you have behaved badly, repent, make what amends you can and address yourself to the task of behaving better next time. On no account brood over your wrongdoing. Rolling in the muck is not the best way of getting clean.")
+        model_select = gr.Dropdown(choices=["S model", "M model", "L model"], label="Select Fine-tuned Model")
+        translate_btn = gr.Button("Translate")
+    
+    with gr.Row():
+        base_output = gr.Textbox(label="Base Model Translation")
+        fine_tuned_output = gr.Textbox(label="Fine-tuned Model Translation")
+    
+    translate_btn.click(translate_text, inputs=[text_input, model_select], outputs=[base_output, fine_tuned_output])
 
 port = int(os.getenv("GRADIO_SERVER_PORT", "7861"))
 demo.launch(server_port=port)