From 0555d4dc75de0afa10d2004df43c0d8898a33252 Mon Sep 17 00:00:00 2001
From: Andres Marafioti <andimarafioti@gmail.com>
Date: Wed, 28 Aug 2024 17:28:25 +0200
Subject: [PATCH] pass the current language around
---
STT/whisper_stt_handler.py | 1 +
TTS/melo_handler.py | 2 +-
s2s_pipeline.py | 1 -
shared_variables.py | 1 +
4 files changed, 3 insertions(+), 2 deletions(-)
create mode 100644 shared_variables.py
diff --git a/STT/whisper_stt_handler.py b/STT/whisper_stt_handler.py
index 0b1ed7e..e964a7c 100644
--- a/STT/whisper_stt_handler.py
+++ b/STT/whisper_stt_handler.py
@@ -8,6 +8,7 @@ import torch
from baseHandler import BaseHandler
from rich.console import Console
import logging
+from shared_variables import current_language
logger = logging.getLogger(__name__)
console = Console()
diff --git a/TTS/melo_handler.py b/TTS/melo_handler.py
index 24e366d..06afcc6 100644
--- a/TTS/melo_handler.py
+++ b/TTS/melo_handler.py
@@ -5,6 +5,7 @@ import librosa
import numpy as np
from rich.console import Console
import torch
+from shared_variables import current_language
logger = logging.getLogger(__name__)
@@ -53,7 +54,6 @@ class MeloTTSHandler(BaseHandler):
def process(self, llm_sentence):
console.print(f"[green]ASSISTANT: {llm_sentence}")
- global current_language
if self.language != current_language:
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]]
diff --git a/s2s_pipeline.py b/s2s_pipeline.py
index d85ade8..2438060 100644
--- a/s2s_pipeline.py
+++ b/s2s_pipeline.py
@@ -48,7 +48,6 @@ os.environ["TORCHINDUCTOR_CACHE_DIR"] = os.path.join(CURRENT_DIR, "tmp")
console = Console()
logging.getLogger("numba").setLevel(logging.WARNING) # quiet down numba logs
-current_language = "en"
def prepare_args(args, prefix):
diff --git a/shared_variables.py b/shared_variables.py
new file mode 100644
index 0000000..649f0f3
--- /dev/null
+++ b/shared_variables.py
@@ -0,0 +1 @@
+current_language = "en"
--
GitLab