Update app.py

app.py

@@ -42,6 +42,7 @@ def infer_tts(ref_audio_orig: str, gen_text: str, speed: float = 1.0, request: g
     try:
         gen_texts = chunk_text(gen_text)
         final_wave_total = None
+        final_sample_rate = 24000
         for i, gen_text in enumerate(gen_texts):
             ref_audio, ref_text = preprocess_ref_audio_text(ref_audio_orig, "")
             final_wave = generate_sentence(
@@ -63,7 +64,7 @@ def infer_tts(ref_audio_orig: str, gen_text: str, speed: float = 1.0, request: g
             spectrogram_path = tmp_spectrogram.name
             save_spectrogram(final_wave_total, spectrogram_path)
 
-        return (final_sample_rate, final_wave), spectrogram_path
+        return (final_sample_rate, final_wave_total), spectrogram_path
     except Exception as e:
         raise gr.Error(f"Error generating voice: {e}")