Update app.py

app.py

@@ -226,23 +226,35 @@ Due to resource limitation, you can only train 10 epochs on maximum.
             )
         
     with gr.Row():
-        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]
-        )
+        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]
+            )
 
     gr.Markdown(load_markdown())