update

app_locally.py

@@ -19,10 +19,9 @@ def predict(prompt, style, audio_file_pth, mic_file_path, use_mic, language):
     text_hint = ''
 
     tts_model = TTS(language=language, device=device)
-    speaker_id = tts_model.hps.data.spk2id
-    speaker_key = speaker_id.lower().replace('_', '-')
-    source_se = torch.load(f'checkpoints/base_speakers/ses/{speaker_key}.pth', map_location=device)
 
+    speaker_key = language.lower().replace('_', '-')
+    source_se = torch.load(f'checkpoints/base_speakers/ses/{speaker_key}.pth', map_location=device)
 
     if use_mic == True:
         if mic_file_path is not None:
@@ -152,7 +151,7 @@ with gr.Blocks(analytics_enabled=False) as demo:
                 value=False,
                 info="Notice: Microphone input may not work properly under traffic",
             )
-            language = gr.Radio(['EN_NEWEST', 'EN', 'ES', 'FR', 'ZH', 'JP', 'KR'], label='Language', value='EN_NEWEST')
+            language = gr.Radio(['EN_NEWEST', 'EN-US', 'EN-BR', 'EN_INDIA', 'EN-AU', 'EN-Default', 'ES', 'FR', 'ZH', 'JP', 'KR'], label='Language', value='EN_NEWEST')
 
             tts_button = gr.Button("Send", elem_id="send-btn", visible=True)