Add spanish

coqui-ai · Nov 15, 2023 · d64cafc · d64cafc
1 parent 2d9525a
commit d64cafc
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/TTS/tts/layers/xtts/tokenizer.py b/TTS/tts/layers/xtts/tokenizer.py
@@ -16,6 +16,7 @@
 from spacy.lang.zh import Chinese
 from spacy.lang.ja import Japanese
 from spacy.lang.ar import Arabic
+from spacy.lang.es import Spanish
 
 
 def get_spacy_lang(lang):
@@ -25,7 +26,10 @@ def get_spacy_lang(lang):
         return Japanese()
     elif lang == "ar":
         return Arabic()
+    elif lang == "es":
+        return Spanish()
     else:
+        # For most languages, Enlish does the job
         return English()
 
 def split_sentence(text, lang, text_split_length=250):