update

app_locally.py

@@ -21,7 +21,6 @@ def predict(prompt, 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)
 
@@ -72,6 +71,9 @@ def predict(prompt, audio_file_pth, mic_file_path, use_mic, language):
     speed = 1.0
 
     #tts_model.tts_to_file(prompt, speaker_id, src_path, speaker=style, language=language)
+    speaker_ids = model.hps.data.spk2id
+    speaker_id = speaker_ids[language]
+
     tts_model.tts_to_file(prompt, speaker_id, src_path)
 
     save_path = f'{output_dir}/output.wav'