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

app.py

@@ -209,7 +209,8 @@ def split_text_into_batches(text, max_chars=200, split_words=SPLIT_WORDS):
         batches.append(current_batch)
     
     return batches
-@spaces.GPU
+
+@gpu_decorator
 def infer_batch(ref_audio, ref_text, gen_text_batches, exp_name, remove_silence, progress=gr.Progress()):
     if exp_name == "F5-TTS":
         ema_model = F5TTS_ema_model
@@ -294,6 +295,7 @@ def infer_batch(ref_audio, ref_text, gen_text_batches, exp_name, remove_silence,
 
     return (target_sample_rate, final_wave), spectrogram_path
 
+@gpu_decorator
 def infer(ref_audio_orig, ref_text, gen_text, exp_name, remove_silence, custom_split_words=''):
     if not custom_split_words.strip():
         custom_words = [word.strip() for word in custom_split_words.split(',')]
@@ -342,7 +344,8 @@ def infer(ref_audio_orig, ref_text, gen_text, exp_name, remove_silence, custom_s
     
     gr.Info(f"Generating audio using {exp_name} in {len(gen_text_batches)} batches")
     return infer_batch((audio, sr), ref_text, gen_text_batches, exp_name, remove_silence)
-    
+
+@gpu_decorator
 def generate_podcast(script, speaker1_name, ref_audio1, ref_text1, speaker2_name, ref_audio2, ref_text2, exp_name, remove_silence):
     # Split the script into speaker blocks
     speaker_pattern = re.compile(f"^({re.escape(speaker1_name)}|{re.escape(speaker2_name)}):", re.MULTILINE)
@@ -678,7 +681,7 @@ with gr.Blocks() as app_emotional:
 
     # Output audio
     audio_output_emotional = gr.Audio(label="Synthesized Audio")
-
+    @gpu_decorator
     def generate_emotional_speech(
         regular_audio,
         regular_ref_text,