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

Merge pull request #70 from tijszwinkels/fix-imports

Browse source 
Fix imports
This commit is contained in:
Dominik Macháček 2024-03-21 00:11:52 +01:00 committed by GitHub
commit 3af93975cc
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