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