Spaces:
Running
Running
Fix kokoro langauge code
Browse files- mcp_host/tts/hf_zero_gpu_tts.py +1 -1
- mcp_host/tts/utils.py +5 -5
mcp_host/tts/hf_zero_gpu_tts.py
CHANGED
|
@@ -56,7 +56,7 @@ def stream_text_to_speech(
|
|
| 56 |
raise ValueError(f"Voice '{voice}' is not available.")
|
| 57 |
|
| 58 |
kokoro_lang = voice[0]
|
| 59 |
-
standard_lang_code = KOKORO_TO_STD_LANG
|
| 60 |
pipe = pipes[kokoro_lang]
|
| 61 |
|
| 62 |
for text in generate_sentences(text_stream, language=standard_lang_code):
|
|
|
|
| 56 |
raise ValueError(f"Voice '{voice}' is not available.")
|
| 57 |
|
| 58 |
kokoro_lang = voice[0]
|
| 59 |
+
standard_lang_code = KOKORO_TO_STD_LANG[kokoro_lang]
|
| 60 |
pipe = pipes[kokoro_lang]
|
| 61 |
|
| 62 |
for text in generate_sentences(text_stream, language=standard_lang_code):
|
mcp_host/tts/utils.py
CHANGED
|
@@ -30,13 +30,13 @@ VOICES = {
|
|
| 30 |
}
|
| 31 |
|
| 32 |
KOKORO_TO_STD_LANG = {
|
| 33 |
-
"a": "en",
|
| 34 |
-
"b": "en",
|
| 35 |
"e": "es",
|
| 36 |
-
"f": "fr",
|
| 37 |
"h": "hi",
|
| 38 |
"i": "it",
|
|
|
|
| 39 |
"j": "ja",
|
| 40 |
-
"p": "pt",
|
| 41 |
"z": "zh",
|
| 42 |
-
}
|
|
|
|
| 30 |
}
|
| 31 |
|
| 32 |
KOKORO_TO_STD_LANG = {
|
| 33 |
+
"a": "en-us",
|
| 34 |
+
"b": "en-gb",
|
| 35 |
"e": "es",
|
| 36 |
+
"f": "fr-fr",
|
| 37 |
"h": "hi",
|
| 38 |
"i": "it",
|
| 39 |
+
"p": "pt-br",
|
| 40 |
"j": "ja",
|
|
|
|
| 41 |
"z": "zh",
|
| 42 |
+
}
|