csukuangfj commited on
Commit
79cd679
1 Parent(s): c145060

add Korean

Browse files
Files changed (1) hide show
  1. model.py +7 -0
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()),