From 006de3e7b044533bd89c0f228a7ce388bed41a51 Mon Sep 17 00:00:00 2001 From: Tijs Zwinkels Date: Wed, 20 Mar 2024 16:02:24 +0100 Subject: [PATCH] Fix imports Now, the ASR implementations do their own imports. No need to import in the factory --- whisper_online.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/whisper_online.py b/whisper_online.py index c90babb..a00547e 100644 --- a/whisper_online.py +++ b/whisper_online.py @@ -58,6 +58,7 @@ class WhisperTimestampedASR(ASRBase): def load_model(self, modelsize=None, cache_dir=None, model_dir=None): import whisper + import whisper_timestamped from whisper_timestamped import transcribe_timestamped self.transcribe_timestamped = transcribe_timestamped if model_dir is not None: @@ -558,10 +559,8 @@ def asr_factory(args, logfile=sys.stderr): asr = OpenaiApiASR(lan=args.lan) else: if backend == "faster-whisper": - from faster_whisper import FasterWhisperASR asr_cls = FasterWhisperASR else: - from whisper_timestamped import WhisperTimestampedASR asr_cls = WhisperTimestampedASR # Only for FasterWhisperASR and WhisperTimestampedASR