alihan/speech-to-speech
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

debug

Browse Source
This commit is contained in:
Andres Marafioti
2024-08-29 10:45:40 +02:00
parent 7b0da0971c
commit d8f7b5fef6
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
TTS/melo_handler.py
View File

@@ -53,11 +53,14 @@ class MeloTTSHandler(BaseHandler):
_ = self.model.tts_to_file("text", self.speaker_id, quiet=True)
def process(self, llm_sentence):
console.print(f"[green]ASSISTANT: {llm_sentence}")
language_id = None
if isinstance(llm_sentence, tuple):
print("llm sentence is tuple!")
llm_sentence, language_id = llm_sentence
console.print(f"[green]ASSISTANT: {llm_sentence}")
if language_id is not None and 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]]