update

app_locally.py

@@ -20,6 +20,7 @@ def predict(prompt, style, audio_file_pth, mic_file_path, use_mic, language):
 
     tts_model = TTS(language=language, device=device)
 
+    speaker_id = language
     speaker_key = language.lower().replace('_', '-')
     source_se = torch.load(f'checkpoints/base_speakers/ses/{speaker_key}.pth', map_location=device)
 
@@ -69,7 +70,7 @@ def predict(prompt, style, audio_file_pth, mic_file_path, use_mic, language):
 
     speed = 1.0
 
-    tts_model.tts(prompt, src_path, speaker=style, language=language)
+    tts_model.tts_to_file(prompt, speaker_id, src_path, speaker=style, language=language)
 
     save_path = f'{output_dir}/output.wav'
     # Run the tone color converter