csukuangfj commited on
Commit
7fcf32c
1 Parent(s): f18e30d

add Czech.

Browse files
Files changed (1) hide show
  1. model.py +8 -0
model.py CHANGED
@@ -273,6 +273,8 @@ def get_pretrained_model(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
273
  return arabic_models[repo_id](repo_id, speed)
274
  elif repo_id in catalan_models:
275
  return catalan_models[repo_id](repo_id, speed)
 
 
276
  else:
277
  raise ValueError(f"Unsupported repo_id: {repo_id}")
278
 
@@ -391,12 +393,18 @@ catalan_models = {
391
  "csukuangfj/vits-piper-ca_ES-upc_pau-x_low": _get_vits_piper,
392
  }
393
 
 
 
 
 
 
394
  language_to_models = {
395
  "English": list(english_models.keys()),
396
  "Chinese (Mandarin, 普通话)": list(chinese_models.keys()),
397
  "Cantonese (粤语)": list(cantonese_models.keys()),
398
  "Arabic": list(arabic_models.keys()),
399
  "Catalan": list(catalan_models.keys()),
 
400
  "French": list(french_models.keys()),
401
  "German": list(german_models.keys()),
402
  "Russian": list(russian_models.keys()),
 
273
  return arabic_models[repo_id](repo_id, speed)
274
  elif repo_id in catalan_models:
275
  return catalan_models[repo_id](repo_id, speed)
276
+ elif repo_id in czech_models:
277
+ return czech_models[repo_id](repo_id, speed)
278
  else:
279
  raise ValueError(f"Unsupported repo_id: {repo_id}")
280
 
 
393
  "csukuangfj/vits-piper-ca_ES-upc_pau-x_low": _get_vits_piper,
394
  }
395
 
396
+ czech_models = {
397
+ "csukuangfj/vits-piper-cs_CZ-jirka-low": _get_vits_piper,
398
+ "csukuangfj/vits-piper-cs_CZ-jirka-medium": _get_vits_piper,
399
+ }
400
+
401
  language_to_models = {
402
  "English": list(english_models.keys()),
403
  "Chinese (Mandarin, 普通话)": list(chinese_models.keys()),
404
  "Cantonese (粤语)": list(cantonese_models.keys()),
405
  "Arabic": list(arabic_models.keys()),
406
  "Catalan": list(catalan_models.keys()),
407
+ "Czech": list(czech_models.keys()),
408
  "French": list(french_models.keys()),
409
  "German": list(german_models.keys()),
410
  "Russian": list(russian_models.keys()),