From 2f266b2d9587aa1967e48cf7f92583c3f94143d8 Mon Sep 17 00:00:00 2001
From: Andres Marafioti <andimarafioti@gmail.com>
Date: Thu, 29 Aug 2024 10:34:10 +0200
Subject: [PATCH] fixy

---
 TTS/melo_handler.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/TTS/melo_handler.py b/TTS/melo_handler.py
index d5cd53c..6b5d831 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()
 
-- 
GitLab