Skip to content

Commit 9829d7c

Browse files
authored
Add two GLaDOS TTS models (#481)
1 parent 80d0192 commit 9829d7c

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

scripts/apk/generate-tts-apk-script.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -79,6 +79,7 @@ def get_piper_models() -> List[TtsModel]:
7979
TtsModel(model_dir="vits-piper-en_GB-southern_english_female-low"),
8080
TtsModel(model_dir="vits-piper-en_GB-sweetbbak-amy"),
8181
TtsModel(model_dir="vits-piper-en_GB-vctk-medium"),
82+
TtsModel(model_dir="vits-piper-en_US-glados"),
8283
TtsModel(model_dir="vits-piper-en_US-amy-low"),
8384
TtsModel(model_dir="vits-piper-en_US-amy-medium"),
8485
TtsModel(model_dir="vits-piper-en_US-arctic-medium"),
@@ -96,6 +97,7 @@ def get_piper_models() -> List[TtsModel]:
9697
TtsModel(model_dir="vits-piper-en_US-ryan-high"),
9798
TtsModel(model_dir="vits-piper-en_US-ryan-low"),
9899
TtsModel(model_dir="vits-piper-en_US-ryan-medium"),
100+
TtsModel(model_dir="vits-piper-es-glados-medium"),
99101
TtsModel(model_dir="vits-piper-es_ES-carlfm-x_low"),
100102
TtsModel(model_dir="vits-piper-es_ES-davefx-medium"),
101103
TtsModel(model_dir="vits-piper-es_ES-mls_10246-low"),
@@ -158,7 +160,7 @@ def get_piper_models() -> List[TtsModel]:
158160
for m in models:
159161
m.data_dir = m.model_dir + "/" + "espeak-ng-data"
160162
m.model_name = m.model_dir[len("vits-piper-") :] + ".onnx"
161-
m.lang = "en"
163+
m.lang = m.model_dir.split("-")[2][:2]
162164

163165
return models
164166

0 commit comments

Comments
 (0)