Fix: отключено логирование VoskAPI
- Модель загружается один раз при старте бота - Логи VoskAPI засоряли вывод - теперь отключены Co-authored-by: Qwen-Coder <qwen-coder@alibabacloud.com>
This commit is contained in:
parent
da3058194f
commit
763b33026a
|
|
@ -21,6 +21,10 @@ class SpeechRecognizer:
|
|||
if not self.enabled:
|
||||
return
|
||||
|
||||
# Отключаем логирование VoskAPI чтобы не засорять логи
|
||||
import logging as py_logging
|
||||
py_logging.getLogger("VoskAPI").setLevel(py_logging.ERROR)
|
||||
|
||||
try:
|
||||
if self.model_name == "vosk":
|
||||
from vosk import Model, KaldiRecognizer
|
||||
|
|
@ -29,16 +33,14 @@ class SpeechRecognizer:
|
|||
# Путь к модели в папке проекта
|
||||
model_path = "./models/vosk/vosk-model-ru-0.22"
|
||||
if not os.path.exists(model_path):
|
||||
logger.warning(f"Модель Vosk не найдена по пути {model_path}")
|
||||
logger.error(f"Модель Vosk не найдена по пути {model_path}")
|
||||
return
|
||||
|
||||
self.model = Model(model_path)
|
||||
logger.warning("Модель Vosk загружена")
|
||||
|
||||
elif self.model_name == "whisper":
|
||||
from faster_whisper import WhisperModel
|
||||
|
||||
logger.warning("Первый запуск Whisper может занять время - скачивается модель (~250MB)")
|
||||
self.model = WhisperModel("small", device="cpu", compute_type="int8")
|
||||
|
||||
except Exception as e:
|
||||
|
|
|
|||
Loading…
Reference in New Issue