Spaces:
Sleeping
Sleeping
csukuangfj
commited on
Commit
•
407c586
1
Parent(s):
9681d61
add cantonese
Browse files
model.py
CHANGED
@@ -203,9 +203,7 @@ def _get_vits_zh_aishell3(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
|
|
203 |
|
204 |
@lru_cache(maxsize=10)
|
205 |
def _get_vits_hf(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
|
206 |
-
|
207 |
-
|
208 |
-
if "fanchen" in repo_id:
|
209 |
model = repo_id.split("/")[-1]
|
210 |
else:
|
211 |
model = repo_id.split("-")[-1]
|
@@ -269,6 +267,10 @@ def get_pretrained_model(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
|
|
269 |
raise ValueError(f"Unsupported repo_id: {repo_id}")
|
270 |
|
271 |
|
|
|
|
|
|
|
|
|
272 |
chinese_models = {
|
273 |
"csukuangfj/vits-zh-hf-theresa": _get_vits_hf,
|
274 |
"csukuangfj/vits-zh-hf-eula": _get_vits_hf,
|
@@ -348,7 +350,8 @@ french_models = {
|
|
348 |
|
349 |
language_to_models = {
|
350 |
"English": list(english_models.keys()),
|
351 |
-
"Chinese": list(chinese_models.keys()),
|
|
|
352 |
"German": list(german_models.keys()),
|
353 |
"Spanish": list(spanish_models.keys()),
|
354 |
"French": list(french_models.keys()),
|
|
|
203 |
|
204 |
@lru_cache(maxsize=10)
|
205 |
def _get_vits_hf(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
|
206 |
+
if "fanchen" in repo_id or "vits-cantonese-hf-xiaomaiiwn" in repo_id:
|
|
|
|
|
207 |
model = repo_id.split("/")[-1]
|
208 |
else:
|
209 |
model = repo_id.split("-")[-1]
|
|
|
267 |
raise ValueError(f"Unsupported repo_id: {repo_id}")
|
268 |
|
269 |
|
270 |
+
chinese_models = {
|
271 |
+
"csukuangfj/vits-cantonese-hf-xiaomaiiwn": _get_vits_hf,
|
272 |
+
}
|
273 |
+
|
274 |
chinese_models = {
|
275 |
"csukuangfj/vits-zh-hf-theresa": _get_vits_hf,
|
276 |
"csukuangfj/vits-zh-hf-eula": _get_vits_hf,
|
|
|
350 |
|
351 |
language_to_models = {
|
352 |
"English": list(english_models.keys()),
|
353 |
+
"Chinese (Mandarin, 普通话)": list(chinese_models.keys()),
|
354 |
+
"Cantonese (粤语)": list(cantonese_models.keys()),
|
355 |
"German": list(german_models.keys()),
|
356 |
"Spanish": list(spanish_models.keys()),
|
357 |
"French": list(french_models.keys()),
|