Update app.py

app.py

@@ -38,20 +38,20 @@ def load_model(model_path, hps):
 
 hps = utils.get_hparams_from_file("configs/vctk_base.json")
 
-# Define a dictionary to store the model paths
+# Define a dictionary to store the model paths for each tab
 model_paths = {
-    "Model 1": "fr_wa_finetuned_pho/G_125000.pth",
-    "Model 2": "fr_wa_finetuned/G_198000.pth",
-    "Model 3": "path_to_model_3_checkpoint.pth",
-    "Model 4": "path_to_model_4_checkpoint.pth"
+    "Phonemes_finetuned": "fr_wa_finetuned_pho/G_125000.pth",
+    "Graphemes_finetuned": "fr_wa_finetuned/G_198000.pth",
+    "Phonemes": "path_to_phonemes_model.pth",
+    "Graphemes": "path_to_graphemes_model.pth"
 }
 
 # Load the initial model
-net_g = load_model(model_paths["Model 1"], hps)
+net_g = load_model(model_paths["Phonemes_finetuned"], hps)
 
-def tts(text, speaker_id, model_choice):
+def tts(text, speaker_id, tab_name):
     global net_g
-    net_g = load_model(model_paths[model_choice], hps)
+    net_g = load_model(model_paths[tab_name], hps)
     
     if len(text) > 2000:
         return "Error: Text is too long", None
@@ -67,18 +67,16 @@ def tts(text, speaker_id, model_choice):
 
 app = gr.Blocks()
 with app:
-    with gr.Tabs():
+    with gr.Tabs() as tabs:
         for tab_name in ["Phonemes_finetuned", "Graphemes_finetuned", "Phonemes", "Graphemes"]:
             with gr.TabItem(tab_name):
-                tts_input1 = gr.TextArea(label="Text in Walloon in phonemes IPA (2000 words limitation)", value="")
+                tts_input1 = gr.TextArea(label="Text in Walloon (2000 words limitation)", value="")
                 tts_input2 = gr.Dropdown(label="Speaker", choices=["Male", "Female"], type="index", value="Male")
-                model_choice = gr.Dropdown(label="Model", choices=list(model_paths.keys()), value="Model 1")
                 tts_submit = gr.Button("Generate", variant="primary")
                 tts_output1 = gr.Textbox(label="Message")
                 tts_output2 = gr.Audio(label="Output")
-                tts_submit.click(tts, [tts_input1, tts_input2, model_choice], [tts_output1, tts_output2])
+                tts_submit.click(tts, [tts_input1, tts_input2], [tts_output1, tts_output2], _js=f"() => '{tab_name}'")
 
     app.launch()
-