Spaces:
Build error
Build error
barghavani
commited on
Commit
·
254f03b
1
Parent(s):
210da08
Update app.py
Browse files
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 |
-
|
14 |
-
|
15 |
-
|
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 |
-
|
20 |
-
|
21 |
-
#["VITS Grapheme Azure (61000)", "checkpoint_61000.pth", "config.json", "saillab/persian-tts-azure-grapheme-60K"],
|
22 |
|
23 |
-
|
24 |
-
|
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 |
-
|
74 |
config_files[model_name] = hf_hub_download(repo_id=repo_name, filename=config_file, use_auth_token=TOKEN)
|
75 |
-
|
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:
|