Spaces:
Sleeping
Sleeping
Commit
·
7fcf32c
1
Parent(s):
f18e30d
add Czech.
Browse files
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()),
|