Commit 5f516671 authored by Administrator's avatar Administrator

debug

parent dfc41839
...@@ -12,10 +12,10 @@ from .type import WhisperMode, SPEECH_ARRAY_INDEX ...@@ -12,10 +12,10 @@ from .type import WhisperMode, SPEECH_ARRAY_INDEX
class Transcribe: class Transcribe:
def __init__(self, args, whisper_model): def __init__(self, args, model):
self.args = args self.args = args
self.sampling_rate = 16000 self.sampling_rate = 16000
self.whisper_model = whisper_model self.whisper_model = model
self.vad_model = None self.vad_model = None
self.detect_speech = None self.detect_speech = None
......
...@@ -140,5 +140,6 @@ def main_args(logger): ...@@ -140,5 +140,6 @@ def main_args(logger):
logger.info(f'load whisper_model: {args.whisper_model} device: {args.device}') logger.info(f'load whisper_model: {args.whisper_model} device: {args.device}')
import whisper import whisper
whisper_model = whisper.load_model(args.whisper_model, args.device) whisper_model = whisper.load_model(args.whisper_model, args.device)
logger.info(f'done.')
return args, whisper_model 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