Update app.py

app.py

@@ -39,7 +39,8 @@ load_dotenv()
 MAX_INPUT_TEXT_LEN = int(os.environ.get("MAX_INPUT_TEXT_LEN", default=500))
 
 # Dynamically read model files, exclude 'speakers.pth'
-model_files = [f for f in os.listdir(os.getcwd()) if f.endswith('.pth') and f != 'speakers.pth']
+# model_files = [f for f in os.listdir(os.getcwd()) if f.endswith('.pth') and f != 'speakers.pth']
+model_files = [f for f in os.listdir('./checkpoints') if f.endswith('.pth')]
 # model_files.sort(key=lambda x: os.path.getmtime(os.path.join(os.getcwd(), x)), reverse=True)
 
 speakers_path = "speakers.pth"
@@ -55,6 +56,7 @@ 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]  #
 
 DEFAULT_SPEAKER_ID = os.environ.get("DEFAULT_SPEAKER_ID", default="pau")
+DEFAULT_CHECKPOINT = os.environ.get("DEFAULT_CHECKPOINT", default=model_files[0])
 model_file = model_files[0]  # change this!!
 
 model_path = os.path.join(os.getcwd(), model_file)
@@ -170,6 +172,7 @@ with gr.Blocks(**AinaGradioTheme().get_kwargs()) as app:
 
             speaker_id = gr.Dropdown(label="Select a voice", choices=speakers_list, value=DEFAULT_SPEAKER_ID,
                                      interactive=True)
+            model_chkpt = gr.Dropdown(label="Select a checkpoint", choices=model_files, value=DEFAULT_CHECKPOINT, interactive=True)
             dataset.change(fn=update_speaker_list, inputs=dataset, outputs=speaker_id)
 
             # model = gr.Dropdown(label="Select a model", choices=model_files, value=DEFAULT_MODEL_FILE_NAME)