model_se

app.py

@@ -65,6 +65,7 @@ def load_model(language):
             language_ids_filename = os.path.basename(config.get("language_ids_file", "language_ids.json"))
             d_vector_filename = os.path.basename(config.get("d_vector_file", ["d_vector.pth"])[0])
             config_se_filename = os.path.basename(config.get("model_args", {}).get("speaker_encoder_config_path", "config_se.json"))
+            model_se_filename = os.path.basename(config.get("model_args", {}).get("speaker_encoder_model_path", "model_se.pth"))
             
             # Download specific model and config files from HuggingFace repo
             model_path = hf_hub_download(repo_id=model_repo, filename=model_name)
@@ -72,6 +73,7 @@ def load_model(language):
             language_ids_file = hf_hub_download(repo_id=model_repo, filename=language_ids_filename)
             d_vector_file = hf_hub_download(repo_id=model_repo, filename=d_vector_filename)
             config_se_file = hf_hub_download(repo_id=model_repo, filename=config_se_filename)
+            model_se_file = hf_hub_download(repo_id=model_repo, filename=model_se_filename)
             
             # Update the config paths to point to the downloaded files
             config["speakers_file"] = speakers_file
@@ -81,6 +83,7 @@ def load_model(language):
             config["model_args"]["language_ids_file"] = language_ids_file
             config["model_args"]["d_vector_file"] = [d_vector_file]
             config["model_args"]["speaker_encoder_config_path"] = config_se_file
+            config["model_args"]["speaker_encoder_model_path"] = model_se_file
             
             # Save the updated config to a temporary file
             import tempfile