admin-valentin/WhisperLiveKit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix imports

Browse source 
Now, the ASR implementations do their own imports. No need to import in the factory
This commit is contained in:
Tijs Zwinkels 2024-03-20 16:02:24 +01:00
parent 8896389ea3
commit 006de3e7b0
1 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
whisper_online.py
View file

@ -58,6 +58,7 @@ class WhisperTimestampedASR(ASRBase):
def load_model(self, modelsize=None, cache_dir=None, model_dir=None): def load_model(self, modelsize=None, cache_dir=None, model_dir=None):
import whisper import whisper
import whisper_timestamped
from whisper_timestamped import transcribe_timestamped from whisper_timestamped import transcribe_timestamped
self.transcribe_timestamped = transcribe_timestamped self.transcribe_timestamped = transcribe_timestamped
if model_dir is not None: if model_dir is not None:
@ -558,10 +559,8 @@ def asr_factory(args, logfile=sys.stderr):
asr = OpenaiApiASR(lan=args.lan) asr = OpenaiApiASR(lan=args.lan)
else: else:
if backend == "faster-whisper": if backend == "faster-whisper":
from faster_whisper import FasterWhisperASR
asr_cls = FasterWhisperASR asr_cls = FasterWhisperASR
else: else:
from whisper_timestamped import WhisperTimestampedASR
asr_cls = WhisperTimestampedASR asr_cls = WhisperTimestampedASR
# Only for FasterWhisperASR and WhisperTimestampedASR # Only for FasterWhisperASR and WhisperTimestampedASR