feat:将transcriber作为环境变量配置

This commit is contained in:
SurfRid3r
2025-04-19 23:13:49 +08:00
parent 2aad103a77
commit 0b7f6ca4ee
5 changed files with 21 additions and 10 deletions

View File

@@ -43,7 +43,7 @@ class NoteGenerator:
def __init__(self):
self.model_size: str = 'base'
self.device: Union[str, None] = None
self.transcriber_type = 'fast-whisper'
self.transcriber_type = os.getenv('TRANSCRIBER_TYPE','fast-whisper')
self.transcriber = self.get_transcriber()
# TODO 需要更换为可调节
@@ -91,7 +91,7 @@ class NoteGenerator:
'''
if self.transcriber_type == 'fast-whisper':
logger.info("使用Whisper")
return get_transcriber()
return get_transcriber(transcriber_type='fast-whisper')
else:
logger.warning("不支持的转义器")
raise ValueError(f"不支持的转义器:{self.transcriber}")