Spaces:
Sleeping
Sleeping
csukuangfj
commited on
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()),
|