Update app.py

app.py

@@ -226,35 +226,23 @@ Due to resource limitation, you can only train 10 epochs on maximum.
             )
         
     with gr.Row():
-        with gr.Column():
-            baseline_btn = gr.Button("Run Baseline", variant="secondary")
-            baseline_btn.click(
-                baseline_model,
-                [
-                    lang,
-                    task,
-                    tempdir_path,
-                ],
-                [ref_text, base_text]
-            )
-        with gr.Column():
-            finetune_btn = gr.Button("Finetune Model", variant="primary")
-            finetune_btn.click(
-                finetune_model,
-                [
-                    lang,
-                    task,
-                    tempdir_path,
-                    log_every,
-                    max_epoch,
-                    scheduler,
-                    warmup_steps,
-                    optimizer,
-                    learning_rate,
-                    weight_decay,
-                ],
-                [trained_model, hyp_text]
-            )
+        finetune_btn = gr.Button("Finetune Model", variant="primary")
+        finetune_btn.click(
+            finetune_model,
+            [
+                lang,
+                task,
+                tempdir_path,
+                log_every,
+                max_epoch,
+                scheduler,
+                warmup_steps,
+                optimizer,
+                learning_rate,
+                weight_decay,
+            ],
+            [trained_model, hyp_text]
+        )
 
     gr.Markdown(load_markdown())