Spaces:
Running
Running
Commit
·
79cd679
1
Parent(s):
c145060
add Korean
Browse files
model.py
CHANGED
|
@@ -348,6 +348,8 @@ def get_pretrained_model(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
|
|
| 348 |
return thai_models[repo_id](repo_id, speed)
|
| 349 |
elif repo_id in persian_models:
|
| 350 |
return persian_models[repo_id](repo_id, speed)
|
|
|
|
|
|
|
| 351 |
else:
|
| 352 |
raise ValueError(f"Unsupported repo_id: {repo_id}")
|
| 353 |
|
|
@@ -644,6 +646,10 @@ persian_models = {
|
|
| 644 |
"csukuangfj/vits-piper-fa-haaniye_low": _get_vits_piper,
|
| 645 |
}
|
| 646 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 647 |
|
| 648 |
language_to_models = {
|
| 649 |
"English": list(english_models.keys()),
|
|
@@ -669,6 +675,7 @@ language_to_models = {
|
|
| 669 |
"Irish": list(irish_models.keys()),
|
| 670 |
"Italian": list(italian_models.keys()),
|
| 671 |
"Kazakh": list(kazakh_models.keys()),
|
|
|
|
| 672 |
"Latvian": list(latvian_models.keys()),
|
| 673 |
"Lithuanian": list(lithuanian_models.keys()),
|
| 674 |
"Luxembourgish": list(luxembourgish_models.keys()),
|
|
|
|
| 348 |
return thai_models[repo_id](repo_id, speed)
|
| 349 |
elif repo_id in persian_models:
|
| 350 |
return persian_models[repo_id](repo_id, speed)
|
| 351 |
+
elif repo_id in korean_models:
|
| 352 |
+
return korean_models[repo_id](repo_id, speed)
|
| 353 |
else:
|
| 354 |
raise ValueError(f"Unsupported repo_id: {repo_id}")
|
| 355 |
|
|
|
|
| 646 |
"csukuangfj/vits-piper-fa-haaniye_low": _get_vits_piper,
|
| 647 |
}
|
| 648 |
|
| 649 |
+
korean_models = {
|
| 650 |
+
"csukuangfj/vits-mimic3-ko_KO-kss_low": _get_vits_piper,
|
| 651 |
+
}
|
| 652 |
+
|
| 653 |
|
| 654 |
language_to_models = {
|
| 655 |
"English": list(english_models.keys()),
|
|
|
|
| 675 |
"Irish": list(irish_models.keys()),
|
| 676 |
"Italian": list(italian_models.keys()),
|
| 677 |
"Kazakh": list(kazakh_models.keys()),
|
| 678 |
+
"Korean": list(korean_models.keys()),
|
| 679 |
"Latvian": list(latvian_models.keys()),
|
| 680 |
"Lithuanian": list(lithuanian_models.keys()),
|
| 681 |
"Luxembourgish": list(luxembourgish_models.keys()),
|