Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -16,7 +16,7 @@ if not IS_DUPLICATE:
|
|
| 16 |
|
| 17 |
CHAR_LIMIT = None if IS_DUPLICATE else 5000
|
| 18 |
models = {gpu: KModel().to('cuda' if gpu else 'cpu').eval() for gpu in [False] + ([True] if CUDA_AVAILABLE else [])}
|
| 19 |
-
pipelines = {lang_code: KPipeline(lang_code=lang_code, model=False) for lang_code in '
|
| 20 |
pipelines['a'].g2p.lexicon.golds['kokoro'] = 'kˈOkəɹO'
|
| 21 |
pipelines['b'].g2p.lexicon.golds['kokoro'] = 'kˈQkəɹQ'
|
| 22 |
|
|
@@ -37,7 +37,7 @@ def generate_first(text, voice='af_heart', speed=1, use_gpu=CUDA_AVAILABLE):
|
|
| 37 |
for _, ps, _ in pipeline_output:
|
| 38 |
print("[ENTER LOOPING]", ps)
|
| 39 |
ref_s = pack[len(ps)-1]
|
| 40 |
-
ps = text
|
| 41 |
try:
|
| 42 |
if use_gpu:
|
| 43 |
audio = forward_gpu(ps, ref_s, speed)
|
|
@@ -76,7 +76,7 @@ def generate_all(text, voice='af_heart', speed=1, use_gpu=CUDA_AVAILABLE):
|
|
| 76 |
first = True
|
| 77 |
for _, ps, _ in pipeline(text, voice, speed):
|
| 78 |
ref_s = pack[len(ps)-1]
|
| 79 |
-
ps = text
|
| 80 |
try:
|
| 81 |
if use_gpu:
|
| 82 |
audio = forward_gpu(ps, ref_s, speed)
|
|
@@ -137,11 +137,11 @@ CHOICES = {
|
|
| 137 |
'🇬🇧 🚹 Fable': 'bm_fable',
|
| 138 |
'🇬🇧 🚹 Lewis': 'bm_lewis',
|
| 139 |
'🇬🇧 🚹 Daniel': 'bm_daniel',
|
| 140 |
-
|
| 141 |
-
|
| 142 |
-
|
| 143 |
-
|
| 144 |
-
|
| 145 |
}
|
| 146 |
for v in CHOICES.values():
|
| 147 |
pipelines[v[0]].load_voice(v)
|
|
|
|
| 16 |
|
| 17 |
CHAR_LIMIT = None if IS_DUPLICATE else 5000
|
| 18 |
models = {gpu: KModel().to('cuda' if gpu else 'cpu').eval() for gpu in [False] + ([True] if CUDA_AVAILABLE else [])}
|
| 19 |
+
pipelines = {lang_code: KPipeline(lang_code=lang_code, model=False) for lang_code in 'abefhip'}
|
| 20 |
pipelines['a'].g2p.lexicon.golds['kokoro'] = 'kˈOkəɹO'
|
| 21 |
pipelines['b'].g2p.lexicon.golds['kokoro'] = 'kˈQkəɹQ'
|
| 22 |
|
|
|
|
| 37 |
for _, ps, _ in pipeline_output:
|
| 38 |
print("[ENTER LOOPING]", ps)
|
| 39 |
ref_s = pack[len(ps)-1]
|
| 40 |
+
ps = text[5:-2]
|
| 41 |
try:
|
| 42 |
if use_gpu:
|
| 43 |
audio = forward_gpu(ps, ref_s, speed)
|
|
|
|
| 76 |
first = True
|
| 77 |
for _, ps, _ in pipeline(text, voice, speed):
|
| 78 |
ref_s = pack[len(ps)-1]
|
| 79 |
+
ps = text[5:-2]
|
| 80 |
try:
|
| 81 |
if use_gpu:
|
| 82 |
audio = forward_gpu(ps, ref_s, speed)
|
|
|
|
| 137 |
'🇬🇧 🚹 Fable': 'bm_fable',
|
| 138 |
'🇬🇧 🚹 Lewis': 'bm_lewis',
|
| 139 |
'🇬🇧 🚹 Daniel': 'bm_daniel',
|
| 140 |
+
'Espanol':'ef_dora',
|
| 141 |
+
'Hindi':'hf_alpha',
|
| 142 |
+
'French':'ff_siwis',
|
| 143 |
+
'Italian':'if_sara',
|
| 144 |
+
'Portuguese':'pf_dora',
|
| 145 |
}
|
| 146 |
for v in CHOICES.values():
|
| 147 |
pipelines[v[0]].load_voice(v)
|