LLM/WhisperLiveKit
1
0
Fork 0
mirror of https://github.com/QuentinFuxa/WhisperLiveKit.git synced 2026-03-07 22:33:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

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
parent 50937bb872 006de3e7b0
commit 3af93975cc
1 changed files with 1 additions and 2 deletions
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):
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