Update finetune.py

finetune.py

@@ -128,10 +128,12 @@ class CustomFinetuneModel(ESPnetS2TModel):
 def finetune_model(lang, task, tempdir_path, log_every, max_epoch, scheduler, warmup_steps, optimizer, learning_rate, weight_decay):
     """Main function for finetuning the model."""
 
-    if not os.path.exists(f"{tempdir_path}/ref.txt"):
-        raise gr.Error("Please run Baseline first.")
+    print("Start generating baseline...")
+    gr.Info("Start generating baseline...")
+    baseline_model(lang, task, tempdir_path)
     
     print("Start loading dataset...")
+    gr.Info("Start Fine-tuning process...")
     if len(tempdir_path) == 0:
         raise gr.Error("Please upload a zip file first.")