Update app.py

app.py

@@ -17,23 +17,29 @@ vocoder = load_vocoder()
 # common usage
 v1_base_cfg = dict(dim=1024, depth=22, heads=16, ff_mult=2, text_dim=512, conv_layers=4)
 v1_small_cfg = dict(dim=768, depth=18, heads=12, ff_mult=2, text_dim=512, conv_layers=4)
+zh_en_vocab_path = str(cached_path("hf://SWivid/F5-TTS/F5TTS_v1_Base/vocab.txt"))
 alg_vocab_path = str(cached_path("hf://chenxie95/F5-TTS_v1_Small_Algerian/vocab.txt"))
 
-tts_model_choice = "v1-base_zh-en"  # default
-tts_model_collections = {
-    "v1-base_zh-en": load_model(
-        DiT,
-        v1_base_cfg,
-        str(cached_path("hf://SWivid/F5-TTS/F5TTS_v1_Base/model_1250000.safetensors")),
-        vocab_file=str(cached_path("hf://SWivid/F5-TTS/F5TTS_v1_Base/vocab.txt")),
-    ),
-    "v1-small_alg-64h_400k": load_model(
-        DiT,
-        v1_small_cfg,
-        str(cached_path("hf://chenxie95/F5-TTS_v1_Small_Algerian/64h_model_400000.safetensors")),
-        vocab_file=alg_vocab_path,
-    ),
+tts_lang_model_collections = {
+    "Mandarin-English": {
+        "v1-base_zh-en": load_model(
+            DiT,
+            v1_base_cfg,
+            str(cached_path("hf://SWivid/F5-TTS/F5TTS_v1_Base/model_1250000.safetensors")),
+            vocab_file=zh_en_vocab_path,
+        ),
+    },
+    "Algerian": {
+        "v1-small_alg-64h_400k": load_model(
+            DiT,
+            v1_small_cfg,
+            str(cached_path("hf://chenxie95/F5-TTS_v1_Small_Algerian/64h_model_400000.safetensors")),
+            vocab_file=alg_vocab_path,
+        ),
+    },
 }
+tts_lang_choice = next(iter(tts_lang_model_collections))  # first as default
+tts_model_choice = next(iter(tts_lang_model_collections[tts_lang_choice]))  # first as default
 
 
 @spaces.GPU
@@ -134,14 +140,33 @@ with gr.Blocks() as demo:
         """
     )
 
-    def switch_tts_model(new_choice):
+    def switch_tts_lang(new_lang_choice):
+        global tts_lang_choice, tts_model_choice
+        tts_lang_choice = new_lang_choice
+        tts_model_choice = next(iter(tts_lang_model_collections[tts_lang_choice]))  # first as default
+        return gr.update(choices=[t for t in tts_lang_model_collections[tts_lang_choice]], value=tts_model_choice)
+
+    def switch_tts_model(new_model_choice):
         global tts_model_choice
-        tts_model_choice = new_choice
+        tts_model_choice = new_model_choice
 
-    choose_tts_model = gr.Dropdown(
-        choices=[t for t in tts_model_collections], label="Choose TTS Model", value=tts_model_choice
-    )
+    with gr.Row():
+        choose_tts_lang = gr.Dropdown(
+            choices=[t for t in tts_lang_model_collections],
+            label="Choose TTS Language",
+            value=tts_lang_choice,
+        )
+        choose_tts_model = gr.Dropdown(
+            choices=[t for t in tts_lang_model_collections[tts_lang_choice]],
+            label="Choose TTS Model",
+            value=tts_model_choice,
+        )
 
+    choose_tts_lang.change(
+        switch_tts_lang,
+        inputs=[choose_tts_lang],
+        outputs=[choose_tts_model],
+    )
     choose_tts_model.change(
         switch_tts_model,
         inputs=[choose_tts_model],