From 712005aff0a19b018b86278a238fd36c5e460abb Mon Sep 17 00:00:00 2001
From: Andres Marafioti <andimarafioti@gmail.com>
Date: Thu, 29 Aug 2024 10:47:48 +0200
Subject: [PATCH] set language

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

diff --git a/TTS/melo_handler.py b/TTS/melo_handler.py
index b881d82..0a86881 100644
--- a/TTS/melo_handler.py
+++ b/TTS/melo_handler.py
@@ -54,7 +54,7 @@ class MeloTTSHandler(BaseHandler):
 
     def process(self, llm_sentence):
         language_id = None
-        
+
         if isinstance(llm_sentence, tuple):
             print("llm sentence is tuple!")
             llm_sentence, language_id = llm_sentence
@@ -62,6 +62,7 @@ class MeloTTSHandler(BaseHandler):
         console.print(f"[green]ASSISTANT: {llm_sentence}")
 
         if language_id is not None and self.language != language_id:
+            self.language = language_id
             self.model = TTS(language=WHISPER_LANGUAGE_TO_MELO_LANGUAGE[self.language], device=self.device)
             self.speaker_id = self.model.hps.data.spk2id[WHISPER_LANGUAGE_TO_MELO_SPEAKER[self.language]]
 
-- 
GitLab