Spaces:
Running
Running
csukuangfj
commited on
Commit
•
117f3c7
1
Parent(s):
9462bda
Add Afrikaans
Browse files
model.py
CHANGED
@@ -355,6 +355,8 @@ def get_pretrained_model(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
|
|
355 |
return persian_models[repo_id](repo_id, speed)
|
356 |
elif repo_id in korean_models:
|
357 |
return korean_models[repo_id](repo_id, speed)
|
|
|
|
|
358 |
else:
|
359 |
raise ValueError(f"Unsupported repo_id: {repo_id}")
|
360 |
|
@@ -648,13 +650,17 @@ thai_models = {
|
|
648 |
}
|
649 |
|
650 |
persian_models = {
|
651 |
-
"csukuangfj/vits-
|
652 |
}
|
653 |
|
654 |
korean_models = {
|
655 |
"csukuangfj/vits-mimic3-ko_KO-kss_low": _get_vits_piper,
|
656 |
}
|
657 |
|
|
|
|
|
|
|
|
|
658 |
|
659 |
language_to_models = {
|
660 |
"English": list(english_models.keys()),
|
@@ -662,6 +668,7 @@ language_to_models = {
|
|
662 |
"Cantonese (粤语)": list(cantonese_models.keys()),
|
663 |
"Min-nan (闽南话)": list(min_nan_models.keys()),
|
664 |
"Arabic": list(arabic_models.keys()),
|
|
|
665 |
"Bengali": list(bengali_models.keys()),
|
666 |
"Bulgarian": list(bulgarian_models.keys()),
|
667 |
"Catalan": list(catalan_models.keys()),
|
|
|
355 |
return persian_models[repo_id](repo_id, speed)
|
356 |
elif repo_id in korean_models:
|
357 |
return korean_models[repo_id](repo_id, speed)
|
358 |
+
elif repo_id in afrikaans_models:
|
359 |
+
return afrikaans_models[repo_id](repo_id, speed)
|
360 |
else:
|
361 |
raise ValueError(f"Unsupported repo_id: {repo_id}")
|
362 |
|
|
|
650 |
}
|
651 |
|
652 |
persian_models = {
|
653 |
+
"csukuangfj/vits-mimic3-fa-haaniye_low": _get_vits_piper,
|
654 |
}
|
655 |
|
656 |
korean_models = {
|
657 |
"csukuangfj/vits-mimic3-ko_KO-kss_low": _get_vits_piper,
|
658 |
}
|
659 |
|
660 |
+
afrikaans_models = {
|
661 |
+
"csukuangfj/vits-mimic3-af_ZA-google-nwu_low": _get_vits_piper,
|
662 |
+
}
|
663 |
+
|
664 |
|
665 |
language_to_models = {
|
666 |
"English": list(english_models.keys()),
|
|
|
668 |
"Cantonese (粤语)": list(cantonese_models.keys()),
|
669 |
"Min-nan (闽南话)": list(min_nan_models.keys()),
|
670 |
"Arabic": list(arabic_models.keys()),
|
671 |
+
"Afrikaans": list(afrikaans_models.keys()),
|
672 |
"Bengali": list(bengali_models.keys()),
|
673 |
"Bulgarian": list(bulgarian_models.keys()),
|
674 |
"Catalan": list(catalan_models.keys()),
|