Fangjun Kuang
Committed by GitHub

Add more piper tts models (#2425)

@@ -143,6 +143,9 @@ jobs: @@ -143,6 +143,9 @@ jobs:
143 vits-piper-fa_IR-ganji-medium 143 vits-piper-fa_IR-ganji-medium
144 vits-piper-fa_IR-ganji_adabi-medium 144 vits-piper-fa_IR-ganji_adabi-medium
145 vits-piper-fa_IR-reza_ibrahim-medium 145 vits-piper-fa_IR-reza_ibrahim-medium
  146 + vits-piper-hi_IN-pratham-medium
  147 + vits-piper-hi_IN-priyamvada-medium
  148 + vits-piper-es_AR-daniela-high
146 ) 149 )
147 for d in ${dirs[@]}; do 150 for d in ${dirs[@]}; do
148 src=scripts/piper/release/$d 151 src=scripts/piper/release/$d
@@ -171,6 +171,7 @@ def get_piper_models() -> List[TtsModel]: @@ -171,6 +171,7 @@ def get_piper_models() -> List[TtsModel]:
171 TtsModel(model_dir="vits-piper-en_US-ryan-high"), 171 TtsModel(model_dir="vits-piper-en_US-ryan-high"),
172 TtsModel(model_dir="vits-piper-en_US-ryan-low"), 172 TtsModel(model_dir="vits-piper-en_US-ryan-low"),
173 TtsModel(model_dir="vits-piper-en_US-ryan-medium"), 173 TtsModel(model_dir="vits-piper-en_US-ryan-medium"),
  174 + TtsModel(model_dir="vits-piper-es_AR-daniela-high"),
174 TtsModel(model_dir="vits-piper-es_ES-carlfm-x_low"), 175 TtsModel(model_dir="vits-piper-es_ES-carlfm-x_low"),
175 TtsModel(model_dir="vits-piper-es_ES-davefx-medium"), 176 TtsModel(model_dir="vits-piper-es_ES-davefx-medium"),
176 TtsModel(model_dir="vits-piper-es_ES-glados-medium"), 177 TtsModel(model_dir="vits-piper-es_ES-glados-medium"),
@@ -194,6 +195,8 @@ def get_piper_models() -> List[TtsModel]: @@ -194,6 +195,8 @@ def get_piper_models() -> List[TtsModel]:
194 TtsModel(model_dir="vits-piper-fr_FR-tjiho-model1"), 195 TtsModel(model_dir="vits-piper-fr_FR-tjiho-model1"),
195 TtsModel(model_dir="vits-piper-fr_FR-tjiho-model2"), 196 TtsModel(model_dir="vits-piper-fr_FR-tjiho-model2"),
196 TtsModel(model_dir="vits-piper-fr_FR-tjiho-model3"), 197 TtsModel(model_dir="vits-piper-fr_FR-tjiho-model3"),
  198 + TtsModel(model_dir="vits-piper-hi_IN-pratham-medium"),
  199 + TtsModel(model_dir="vits-piper-hi_IN-priyamvada-medium"),
197 TtsModel(model_dir="vits-piper-hu_HU-anna-medium"), 200 TtsModel(model_dir="vits-piper-hu_HU-anna-medium"),
198 TtsModel(model_dir="vits-piper-hu_HU-berta-medium"), 201 TtsModel(model_dir="vits-piper-hu_HU-berta-medium"),
199 TtsModel(model_dir="vits-piper-hu_HU-imre-medium"), 202 TtsModel(model_dir="vits-piper-hu_HU-imre-medium"),
@@ -254,13 +254,21 @@ def get_es_models(): @@ -254,13 +254,21 @@ def get_es_models():
254 PiperModel(name="claude", kind="high", sr=22050, ns=1), 254 PiperModel(name="claude", kind="high", sr=22050, ns=1),
255 ] 255 ]
256 256
  257 + # Argentina
  258 + es_AR = [
  259 + PiperModel(name="daniela", kind="high", sr=22050, ns=1),
  260 + ]
  261 +
257 for m in es_ES: 262 for m in es_ES:
258 m.lang = "es_ES" 263 m.lang = "es_ES"
259 264
260 for m in es_MX: 265 for m in es_MX:
261 m.lang = "es_MX" 266 m.lang = "es_MX"
262 267
263 - ans = es_ES + es_MX 268 + for m in es_AR:
  269 + m.lang = "es_AR"
  270 +
  271 + ans = es_ES + es_MX + es_AR
264 272
265 for m in ans: 273 for m in ans:
266 if m.model_name == "": 274 if m.model_name == "":
@@ -392,6 +400,35 @@ def get_fr_models(): @@ -392,6 +400,35 @@ def get_fr_models():
392 return ans 400 return ans
393 401
394 402
  403 +# hindi
  404 +def get_hi_models():
  405 + hi_IN = [
  406 + PiperModel(name="pratham", kind="medium", sr=22050, ns=1),
  407 + PiperModel(name="priyamvada", kind="medium", sr=22050, ns=1),
  408 + ]
  409 +
  410 + for m in hi_IN:
  411 + m.lang = "hi_IN"
  412 + if m.model_name == "":
  413 + m.model_name = f"{m.lang}-{m.name}-{m.kind}.onnx"
  414 +
  415 + ans = hi_IN
  416 +
  417 + for m in ans:
  418 + m.text = "यह मत पूछो कि तुम्हारा देश तुम्हारे लिए क्या कर सकता है। यह पूछो कि तुम अपने देश के लिए क्या कर सकते हो।"
  419 + code = m.lang[:2]
  420 + if m.cmd == "":
  421 + m.cmd = f"""
  422 + wget -qq https://huggingface.co/rhasspy/piper-voices/resolve/main/{code}/{m.lang}/{m.name}/{m.kind}/{m.model_name}
  423 + wget -qq https://huggingface.co/rhasspy/piper-voices/resolve/main/{code}/{m.lang}/{m.name}/{m.kind}/{m.model_name}.json
  424 + wget -qq https://huggingface.co/rhasspy/piper-voices/resolve/main/{code}/{m.lang}/{m.name}/{m.kind}/MODEL_CARD
  425 + """
  426 + if m.url == "":
  427 + m.url = f"https://huggingface.co/rhasspy/piper-voices/tree/main/{code}/{m.lang}/{m.name}/{m.kind}"
  428 +
  429 + return ans
  430 +
  431 +
395 # hungarian 432 # hungarian
396 def get_hu_models(): 433 def get_hu_models():
397 hu_HU = [ 434 hu_HU = [
@@ -1411,6 +1448,7 @@ def get_all_models(): @@ -1411,6 +1448,7 @@ def get_all_models():
1411 ans += get_fa_models() 1448 ans += get_fa_models()
1412 ans += get_fi_models() 1449 ans += get_fi_models()
1413 ans += get_fr_models() 1450 ans += get_fr_models()
  1451 + ans += get_hi_models()
1414 ans += get_hu_models() 1452 ans += get_hu_models()
1415 ans += get_is_models() 1453 ans += get_is_models()
1416 ans += get_it_models() 1454 ans += get_it_models()