Spaces:
Running
Running
csukuangfj
commited on
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 |
|