Skip to content
Snippets Groups Projects
Commit 712005af authored by Andres Marafioti's avatar Andres Marafioti
Browse files

set language

...@@ -54,7 +54,7 @@ class MeloTTSHandler(BaseHandler): ...@@ -54,7 +54,7 @@ class MeloTTSHandler(BaseHandler):
def process(self, llm_sentence): def process(self, llm_sentence):
language_id = None language_id = None
if isinstance(llm_sentence, tuple): if isinstance(llm_sentence, tuple):
print("llm sentence is tuple!") print("llm sentence is tuple!")
llm_sentence, language_id = llm_sentence llm_sentence, language_id = llm_sentence
...@@ -62,6 +62,7 @@ class MeloTTSHandler(BaseHandler): ...@@ -62,6 +62,7 @@ class MeloTTSHandler(BaseHandler):
console.print(f"[green]ASSISTANT: {llm_sentence}") console.print(f"[green]ASSISTANT: {llm_sentence}")
if language_id is not None and self.language != language_id: 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.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]] self.speaker_id = self.model.hps.data.spk2id[WHISPER_LANGUAGE_TO_MELO_SPEAKER[self.language]]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment