barghavani commited on
Commit
7ca6568
·
1 Parent(s): b01e728

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -8,7 +8,7 @@ from huggingface_hub import hf_hub_download
8
  MODEL_INFO = [
9
  ["vits-multispeaker-495586", "best_model_495586.pth", "config.json", "saillab/vits_multi_cv_15_validated_dataset/resolve/main/","speakers.pth"],
10
  #["VITS Grapheme Multispeaker CV15(reduct)(best at 17864)", "best_model_17864.pth", "config.json", "saillab/persian-tts-cv15-reduct-grapheme-multispeaker"]
11
- ["VITS Grapheme Azure (61000)", "checkpoint_61000.pth", "config.json", "saillab/persian-tts-azure-grapheme-60K"]
12
  ]
13
 
14
  # # Extract model names from MODEL_INFO
@@ -31,8 +31,8 @@ TOKEN = os.getenv('HUGGING_FACE_HUB_TOKEN')
31
  # hf_hub_download(repo_id=repo_name, filename=model_file, use_auth_token=TOKEN)
32
  # hf_hub_download(repo_id=repo_name, filename=config_file, use_auth_token=TOKEN)
33
 
34
- repo_name = "saillab/persian-tts-azure-grapheme-60K"
35
- filename = "checkpoint_61000.pth"
36
 
37
  model_file = hf_hub_download(repo_name, filename, use_auth_token=TOKEN)
38
  config_file = hf_hub_download(repo_name, "config.json", use_auth_token=TOKEN)
 
8
  MODEL_INFO = [
9
  ["vits-multispeaker-495586", "best_model_495586.pth", "config.json", "saillab/vits_multi_cv_15_validated_dataset/resolve/main/","speakers.pth"],
10
  #["VITS Grapheme Multispeaker CV15(reduct)(best at 17864)", "best_model_17864.pth", "config.json", "saillab/persian-tts-cv15-reduct-grapheme-multispeaker"]
11
+ #["VITS Grapheme Azure (61000)", "checkpoint_61000.pth", "config.json", "saillab/persian-tts-azure-grapheme-60K"]
12
  ]
13
 
14
  # # Extract model names from MODEL_INFO
 
31
  # hf_hub_download(repo_id=repo_name, filename=model_file, use_auth_token=TOKEN)
32
  # hf_hub_download(repo_id=repo_name, filename=config_file, use_auth_token=TOKEN)
33
 
34
+ repo_name = "saillab/vits_multi_cv_15_validated_dataset/resolve/main/"
35
+ filename = "cbest_model_495586.pth"
36
 
37
  model_file = hf_hub_download(repo_name, filename, use_auth_token=TOKEN)
38
  config_file = hf_hub_download(repo_name, "config.json", use_auth_token=TOKEN)