Spaces:
Running
Running
Commit
·
bd7955f
1
Parent(s):
2978c02
add more english models
Browse files
model.py
CHANGED
|
@@ -166,14 +166,14 @@ def _get_vits_piper_de_DE(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
|
|
| 166 |
|
| 167 |
|
| 168 |
@lru_cache(maxsize=10)
|
| 169 |
-
def
|
| 170 |
-
repo_id: str, speed: float
|
| 171 |
-
) -> sherpa_onnx.OfflineTts:
|
| 172 |
assert repo_id == "csukuangfj/vits-piper-en_US-lessac-medium"
|
|
|
|
|
|
|
| 173 |
|
| 174 |
model = get_file(
|
| 175 |
repo_id=repo_id,
|
| 176 |
-
filename="
|
| 177 |
subfolder=".",
|
| 178 |
)
|
| 179 |
|
|
@@ -265,7 +265,23 @@ chinese_models = {
|
|
| 265 |
|
| 266 |
english_models = {
|
| 267 |
"csukuangfj/vits-vctk": _get_vits_vctk,
|
| 268 |
-
"csukuangfj/vits-piper-en_US-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 269 |
"csukuangfj/vits-ljs": _get_vits_ljs,
|
| 270 |
}
|
| 271 |
|
|
|
|
| 166 |
|
| 167 |
|
| 168 |
@lru_cache(maxsize=10)
|
| 169 |
+
def _get_vits_piper_en_US(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
|
|
|
|
|
|
|
| 170 |
assert repo_id == "csukuangfj/vits-piper-en_US-lessac-medium"
|
| 171 |
+
n = len("vits-piper-")
|
| 172 |
+
name = repo_id.split("/")[1][n:]
|
| 173 |
|
| 174 |
model = get_file(
|
| 175 |
repo_id=repo_id,
|
| 176 |
+
filename=f"{name}.onnx",
|
| 177 |
subfolder=".",
|
| 178 |
)
|
| 179 |
|
|
|
|
| 265 |
|
| 266 |
english_models = {
|
| 267 |
"csukuangfj/vits-vctk": _get_vits_vctk,
|
| 268 |
+
"csukuangfj/vits-piper-en_US-amy-low": _get_vits_piper_en_US,
|
| 269 |
+
"csukuangfj/vits-piper-en_US-amy-medium": _get_vits_piper_en_US,
|
| 270 |
+
"csukuangfj/vits-piper-en_US-arctic-medium": _get_vits_piper_en_US,
|
| 271 |
+
"csukuangfj/vits-piper-en_US-danny-low": _get_vits_piper_en_US,
|
| 272 |
+
"csukuangfj/vits-piper-en_US-hfc_male-medium": _get_vits_piper_en_US,
|
| 273 |
+
"csukuangfj/vits-piper-en_US-joe-medium": _get_vits_piper_en_US,
|
| 274 |
+
"csukuangfj/vits-piper-en_US-kathleen-low": _get_vits_piper_en_US,
|
| 275 |
+
"csukuangfj/vits-piper-en_US-kusal-medium": _get_vits_piper_en_US,
|
| 276 |
+
"csukuangfj/vits-piper-en_US-l2arctic-medium": _get_vits_piper_en_US,
|
| 277 |
+
"csukuangfj/vits-piper-en_US-lessac-low": _get_vits_piper_en_US,
|
| 278 |
+
"csukuangfj/vits-piper-en_US-lessac-medium": _get_vits_piper_en_US,
|
| 279 |
+
"csukuangfj/vits-piper-en_US-lessac-high": _get_vits_piper_en_US,
|
| 280 |
+
"csukuangfj/vits-piper-en_US-libritts-high": _get_vits_piper_en_US,
|
| 281 |
+
"csukuangfj/vits-piper-en_US-libritts_r-medium": _get_vits_piper_en_US,
|
| 282 |
+
"csukuangfj/vits-piper-en_US-ryan-low": _get_vits_piper_en_US,
|
| 283 |
+
"csukuangfj/vits-piper-en_US-ryan-medium": _get_vits_piper_en_US,
|
| 284 |
+
"csukuangfj/vits-piper-en_US-ryan-high": _get_vits_piper_en_US,
|
| 285 |
"csukuangfj/vits-ljs": _get_vits_ljs,
|
| 286 |
}
|
| 287 |
|