Update tts.py

tts.py

@@ -20,6 +20,13 @@ from vits import commons, utils
 from vits.models import SynthesizerTrn
 
 
+TTS_LANGUAGES = {}
+with open(f"data/tts/all_langs.tsv") as f:
+    for line in f:
+        iso, name = line.split(" ", 1)
+        TTS_LANGUAGES[iso] = name
+
+
 class TextMapper(object):
     def __init__(self, vocab_file):
         self.symbols = [
@@ -84,12 +91,11 @@ class TextMapper(object):
         return text
 
 
-def synthesize(text, lang, speed):
-
+def synthesize(text, lang, speed=None):
     if speed is None:
         speed = 1.0
 
-    lang_code = lang.split(":")[0].strip()
+    lang_code = lang.split()[0].strip()
 
     vocab_file = hf_hub_download(
         repo_id="facebook/mms-tts",
@@ -165,9 +171,9 @@ def synthesize(text, lang, speed):
 
 
 TTS_EXAMPLES = [
-    ["I am going to the store.", "eng: English"],
-    ["안녕하세요", "kor: Korean"],
-    ["क्या मुझे पीने का पानी मिल सकता है?", "hin: Hindi"],
-    ["Tanış olmağıma çox şadam", "azj-script_latin: Azerbaijani, North"],
-    ["Mu zo murna a cikin ƙasar.", "hau: Hausa"],
+    ["I am going to the store.", "eng (English)"],
+    ["안녕하세요.", "kor (Korean)"],
+    ["क्या मुझे पीने का पानी मिल सकता है?", "hin (Hindi)"],
+    ["Tanış olmağıma çox şadam", "azj-script_latin (Azerbaijani, North)"],
+    ["Mu zo murna a cikin ƙasar.", "hau (Hausa)"],
 ]