barghavani commited on
Commit
254f03b
·
1 Parent(s): 210da08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -11
app.py CHANGED
@@ -10,18 +10,16 @@ import json
10
  # Define constants
11
  MODEL_INFO = [
12
  #["vits checkpoint 57000", "checkpoint_57000.pth", "config.json", "mhrahmani/persian-tts-vits-0"],
13
- # ["VITS Grapheme Multispeaker CV15(reduct)(best at 17864)", "best_model_17864.pth", "config.json",
14
- # "saillab/persian-tts-cv15-reduct-grapheme-multispeaker"],
15
- #["VITS Grapheme Multispeaker CV15(reduct)(22000)", "checkpoint_22000.pth", "config.json", "saillab/persian-tts-cv15-reduct-grapheme-multispeaker", "speakers.pth"],
16
- #["VITS Grapheme Multispeaker CV15(reduct)(26000)", "checkpoint_25000.pth", "config.json", "saillab/persian-tts-cv15-reduct-grapheme-multispeaker", "speakers.pth"],
17
  ["vits-multispeaker-retrain-train set-90K", "best_model_56960.pth", "config.json", "saillab/multi_speaker","speakers.pth"],
18
 
19
- # ["VITS Grapheme Azure (best at 15934)", "best_model_15934.pth", "config.json",
20
- # "saillab/persian-tts-azure-grapheme-60K"],
21
- #["VITS Grapheme Azure (61000)", "checkpoint_61000.pth", "config.json", "saillab/persian-tts-azure-grapheme-60K"],
22
 
23
- #["VITS Grapheme ARM24 Fine-Tuned on 1 (66651)", "best_model_66651.pth", "config.json","saillab/persian-tts-grapheme-arm24-finetuned-on1"],
24
- #["VITS Grapheme ARM24 Fine-Tuned on 1 (120000)", "checkpoint_120000.pth", "config.json","saillab/persian-tts-grapheme-arm24-finetuned-on1"],
25
 
26
  # ... Add other models similarly
27
  ]
@@ -70,9 +68,9 @@ for info in MODEL_INFO:
70
 
71
  # Download model and config files
72
  model_files[model_name] = hf_hub_download(repo_id=repo_name, filename=model_file, use_auth_token=TOKEN)
73
- #model_files[model_name] = hf_hub_download(repo_id=repo_name, filename=model_file)
74
  config_files[model_name] = hf_hub_download(repo_id=repo_name, filename=config_file, use_auth_token=TOKEN)
75
- #config_files[model_name] = hf_hub_download(repo_id=repo_name, filename=config_file)
76
 
77
  # Download speakers.pth if it exists
78
  if speaker_file:
 
10
  # Define constants
11
  MODEL_INFO = [
12
  #["vits checkpoint 57000", "checkpoint_57000.pth", "config.json", "mhrahmani/persian-tts-vits-0"],
13
+ ["VITS Grapheme Multispeaker CV15(reduct)(best at 17864)", "best_model_17864.pth", "config.json", "saillab/persian-tts-cv15-reduct-grapheme-multispeaker"],
14
+ ["VITS Grapheme Multispeaker CV15(reduct)(22000)", "checkpoint_22000.pth", "config.json", "saillab/persian-tts-cv15-reduct-grapheme-multispeaker", "speakers.pth"],
15
+ ["VITS Grapheme Multispeaker CV15(reduct)(26000)", "checkpoint_25000.pth", "config.json", "saillab/persian-tts-cv15-reduct-grapheme-multispeaker", "speakers.pth"],
 
16
  ["vits-multispeaker-retrain-train set-90K", "best_model_56960.pth", "config.json", "saillab/multi_speaker","speakers.pth"],
17
 
18
+ ["VITS Grapheme Azure (best at 15934)", "best_model_15934.pth", "config.json","saillab/persian-tts-azure-grapheme-60K"],
19
+ ["VITS Grapheme Azure (61000)", "checkpoint_61000.pth", "config.json", "saillab/persian-tts-azure-grapheme-60K"],
 
20
 
21
+ ["VITS Grapheme ARM24 Fine-Tuned on 1 (66651)", "best_model_66651.pth", "config.json","saillab/persian-tts-grapheme-arm24-finetuned-on1"],
22
+ ["VITS Grapheme ARM24 Fine-Tuned on 1 (120000)", "checkpoint_120000.pth", "config.json","saillab/persian-tts-grapheme-arm24-finetuned-on1"],
23
 
24
  # ... Add other models similarly
25
  ]
 
68
 
69
  # Download model and config files
70
  model_files[model_name] = hf_hub_download(repo_id=repo_name, filename=model_file, use_auth_token=TOKEN)
71
+
72
  config_files[model_name] = hf_hub_download(repo_id=repo_name, filename=config_file, use_auth_token=TOKEN)
73
+
74
 
75
  # Download speakers.pth if it exists
76
  if speaker_file: