sitatech commited on
Commit
1e3e7b5
·
1 Parent(s): 04e6357

Fix kokoro langauge code

Browse files
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.get(kokoro_lang, "en")
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
+ }