Commit 5f516671 authored by Administrator's avatar Administrator

debug

parent dfc41839
......@@ -12,10 +12,10 @@ from .type import WhisperMode, SPEECH_ARRAY_INDEX
class Transcribe:
def __init__(self, args, whisper_model):
def __init__(self, args, model):
self.args = args
self.sampling_rate = 16000
self.whisper_model = whisper_model
self.whisper_model = model
self.vad_model = None
self.detect_speech = None
......
......@@ -140,5 +140,6 @@ def main_args(logger):
logger.info(f'load whisper_model: {args.whisper_model} device: {args.device}')
import whisper
whisper_model = whisper.load_model(args.whisper_model, args.device)
logger.info(f'done.')
return args, whisper_model
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment