Update app.py

app.py

@@ -26,7 +26,11 @@ models = {
     'KR': TTS(language='KR', device=device),
 }
 
-speaker_ids = models['EN'].hps.data.spk2id
+all_speaker_ids = []
+for m in models.values():
+    all_speaker_ids.extend(list(m.hps.data.spk2id.keys()))
+# 重複を消してリスト化
+all_speaker_ids = list(set(all_speaker_ids))
 
 default_text_dict = {
     'EN': 'The field of text-to-speech has seen rapid development recently.',
@@ -60,7 +64,7 @@ def load_speakers(language, text):
 with gr.Blocks() as demo:
     gr.Markdown('# MeloTTS Demo')
     with gr.Group():
-        speaker = gr.Dropdown(speaker_ids.keys(), interactive=True, value='EN-US', label='Speaker')
+        speaker = gr.Dropdown(choices=all_speaker_ids, interactive=True, value='EN-US', label='Speaker')
         language = gr.Radio(['EN', 'ES', 'FR', 'ZH', 'JP', 'KR'], label='Language', value='EN')
         speed = gr.Slider(label='Speed', minimum=0.1, maximum=10.0, value=1.0, interactive=True, step=0.1)
         text = gr.Textbox(label="Text to speak", value=default_text_dict['EN'])