Tijs Zwinkels commited on
Commit
006de3e
·
1 Parent(s): 8896389

Fix imports

Browse files

Now, the ASR implementations do their own imports. No need to import in the factory

Files changed (1) hide show
  1. whisper_online.py +1 -2
whisper_online.py CHANGED
@@ -58,6 +58,7 @@ class WhisperTimestampedASR(ASRBase):
58
 
59
  def load_model(self, modelsize=None, cache_dir=None, model_dir=None):
60
  import whisper
 
61
  from whisper_timestamped import transcribe_timestamped
62
  self.transcribe_timestamped = transcribe_timestamped
63
  if model_dir is not None:
@@ -558,10 +559,8 @@ def asr_factory(args, logfile=sys.stderr):
558
  asr = OpenaiApiASR(lan=args.lan)
559
  else:
560
  if backend == "faster-whisper":
561
- from faster_whisper import FasterWhisperASR
562
  asr_cls = FasterWhisperASR
563
  else:
564
- from whisper_timestamped import WhisperTimestampedASR
565
  asr_cls = WhisperTimestampedASR
566
 
567
  # Only for FasterWhisperASR and WhisperTimestampedASR
 
58
 
59
  def load_model(self, modelsize=None, cache_dir=None, model_dir=None):
60
  import whisper
61
+ import whisper_timestamped
62
  from whisper_timestamped import transcribe_timestamped
63
  self.transcribe_timestamped = transcribe_timestamped
64
  if model_dir is not None:
 
559
  asr = OpenaiApiASR(lan=args.lan)
560
  else:
561
  if backend == "faster-whisper":
 
562
  asr_cls = FasterWhisperASR
563
  else:
 
564
  asr_cls = WhisperTimestampedASR
565
 
566
  # Only for FasterWhisperASR and WhisperTimestampedASR