Sync from GitHub repo

This Space is synced from the GitHub repo: https://github.com/SWivid/F5-TTS. Please submit contributions to the Space there

finetune_gradio.py

@@ -254,6 +254,7 @@ def start_training(
         del tts_api
         gc.collect()
         torch.cuda.empty_cache()
+        tts_api = None
 
     path_project = os.path.join(path_data, dataset_name + "_pinyin")
 
@@ -698,7 +699,7 @@ def get_random_sample_infer(project_name):
     )
 
 
-def infer(project_name, file_checkpoint, exp_name, ref_text, ref_audio, gen_text, nfe_step):
+def infer(file_checkpoint, exp_name, ref_text, ref_audio, gen_text, nfe_step):
     global last_checkpoint, last_device, tts_api
 
     if not os.path.isfile(file_checkpoint):
@@ -917,7 +918,7 @@ with gr.Blocks() as app:
 
             check_button_infer.click(
                 fn=infer,
-                inputs=[project_name, file_checkpoint_pt, exp_name, ref_text, ref_audio, gen_text, nfe_step],
+                inputs=[file_checkpoint_pt, exp_name, ref_text, ref_audio, gen_text, nfe_step],
                 outputs=[gen_audio],
             )