diff --git a/TTS/melo_handler.py b/TTS/melo_handler.py index d5cd53ce18c25906d2ba2d264f76633803cb7c83..6b5d83147e16e6c1fdf87c3d0f45daffc09f1aa0 100644 --- a/TTS/melo_handler.py +++ b/TTS/melo_handler.py @@ -43,8 +43,8 @@ class MeloTTSHandler(BaseHandler): self.should_listen = should_listen self.device = device self.language = "<|" + language + "|>" # 'Tokenize' the language code to do less operations - self.model = TTS(language=WHISPER_LANGUAGE_TO_MELO_LANGUAGE[language], device=device) - self.speaker_id = self.model.hps.data.spk2id[WHISPER_LANGUAGE_TO_MELO_SPEAKER[speaker_to_id]] + self.model = TTS(language=WHISPER_LANGUAGE_TO_MELO_LANGUAGE[self.language], device=device) + self.speaker_id = self.model.hps.data.spk2id[WHISPER_LANGUAGE_TO_MELO_SPEAKER["<|" + speaker_to_id + "|>"]] self.blocksize = blocksize self.warmup()