mirror of
https://github.com/JefferyHcool/BiliNote.git
synced 2026-05-11 18:10:06 +08:00
1ebf236f4f
- 新增 logger 模块，用于全局日志记录 - 在关键位置添加日志输出，包括事件处理、文件清理、FFmpeg 检查等 - 优化数据库操作，增加日志记录 - 在主程序启动时添加日志输出
19 lines
742 B
Python
19 lines
742 B
Python
from app.transcriber.whisper import WhisperTranscriber
|
||
from app.utils.logger import get_logger
|
||
logger = get_logger(__name__)
|
||
|
||
logger.info('实例化transcriber')
|
||
# TODO:后面需要加入逻辑选择
|
||
_transcriber = None
|
||
|
||
def get_transcriber(model_size="base", device="cuda"):
|
||
global _transcriber
|
||
|
||
if _transcriber is None:
|
||
logger.info('不存在 transcriber ，开始实例化transcriber。')
|
||
try:
|
||
_transcriber = WhisperTranscriber(model_size=model_size, device=device)
|
||
logger.info(f'实例化transcriber成功。参数：{model_size}, {device} ')
|
||
except Exception as e:
|
||
logger.error(f"实例化transcriber失败，请检查是否安装whisper。{e}")
|
||
return _transcriber |