Update app.py

app.py

@@ -55,15 +55,21 @@ festcat_speakers = [s for s in speakers_list if len(s) == 3]  #
 google_speakers = [s for s in speakers_list if 3 < len(s) < 20]  #
 commonvoice_speakers = [s for s in speakers_list if len(s) > 20]  #
 
+hop_128_checkpoints = [c for c in model_files if c.split('_')[1] == "M"]
+hop_96_checkpoints = [c for c in model_files if c.split('_')[1] == "reduced"]
+
 DEFAULT_SPEAKER_ID = os.environ.get("DEFAULT_SPEAKER_ID", default="pau")
-DEFAULT_CHECKPOINT = os.environ.get("DEFAULT_CHECKPOINT", default=model_files[0])
+DEFAULT_CHECKPOINT = os.environ.get("DEFAULT_CHECKPOINT", default=model_files[-1])
+
+model_config = "config.json"  # by default 128 hop
+
 # model_file = model_files[0]  # change this!!
 
 # model_path = os.path.join(os.getcwd(), model_file)
 # config_path = os.path.join(os.getcwd(), "config.json")
 
-vocoder_path = None
-vocoder_config_path = None
+# vocoder_path = None
+# vocoder_config_path = None
 
 # synthesizer = Synthesizer(
 #     model_path, config_path, speakers_path, None, vocoder_path, vocoder_config_path,
@@ -89,7 +95,12 @@ def tts_inference(text: str, speaker_idx: str = None, model_file: str=None):
 
     model_path = os.path.join(os.getcwd(), model_file)
     speakers_file_path = "speakers.pth"
-    config_path = "config.json"
+    if model_file.split('_')[1] == "M":
+        config_path = "config.json"
+    elif model_file.split('_')[1] == "reduced":
+        config_path = "config_hop_96.json"
+    else:
+        config_path = "config.json"
     vocoder_path = None
     vocoder_config_path = None
 
@@ -179,11 +190,30 @@ with gr.Blocks(**AinaGradioTheme().get_kwargs()) as app:
                 return gr.update(choices=current_speakers, value=current_speakers[0])
 
 
+            def update_checkpoint_list(model_hop):
+                print("Updating checkpoint list based on model config:", model_hop)
+                if model_hop == "hop_size_128":
+                    current_checkpoints = hop_128_checkpoints
+                    # model_config = "config.json"
+                elif model_hop == "hop_size_96":
+                    current_checkpoints = hop_96_checkpoints
+                else:
+                    current_checkpoints = model_files
+
+                return gr.update(choices=current_checkpoints, value=current_checkpoints[0])
+                
+
+
             speaker_id = gr.Dropdown(label="Select a voice", choices=speakers_list, value=DEFAULT_SPEAKER_ID,
                                      interactive=True)
             dataset.change(fn=update_speaker_list, inputs=dataset, outputs=speaker_id)
 
-            model_chkpt = gr.Dropdown(label="Select a checkpoint", choices=model_files, value=DEFAULT_CHECKPOINT, interactive=True)
+            model_hop = gr.Radio(["hop_size_128", "hop_size_96"], label="Model Type", value="hop_size_128")
+
+            model_chkpt = gr.Dropdown(label="Select a checkpoint", choices=model_files, value=DEFAULT_CHECKPOINT,
+                                      interactive=True)
+
+            model_hop.change(fn=update_checkpoint_list, inputs=model_hop, outputs=model_chkpt)
 
             # model = gr.Dropdown(label="Select a model", choices=model_files, value=DEFAULT_MODEL_FILE_NAME)
             with gr.Row():