update

app_locally.py

@@ -19,7 +19,10 @@ def predict(prompt, audio_file_pth, mic_file_path, use_mic, language):
     # initialize a empty info
     text_hint = ''
 
-    tts_model = TTS(language=language, device=device)
+    lang_code = language
+    if language.startswith("EN"):
+        lang_code = "EN"
+    tts_model = TTS(language=lang_code, device=device)
 
     speaker_key = language.lower().replace('_', '-')
     source_se = torch.load(f'checkpoints/base_speakers/ses/{speaker_key}.pth', map_location=device)