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

import WhisperModel

Browse source
This commit is contained in:
Dominik Macháček 2023-05-25 14:28:18 +02:00
parent a1ba5e6c3a
commit 33369a9555
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
whisper_online.py
View file

@ -6,6 +6,7 @@ from functools import lru_cache
import time import time
from mosestokenizer import MosesTokenizer from mosestokenizer import MosesTokenizer
@lru_cache @lru_cache
def load_audio(fname): def load_audio(fname):
a, _ = librosa.load(fname, sr=16000) a, _ = librosa.load(fname, sr=16000)
@ -89,6 +90,8 @@ class FasterWhisperASR(ASRBase):
sep = "" sep = ""
def load_model(self, modelsize=None, cache_dir=None, model_dir=None): def load_model(self, modelsize=None, cache_dir=None, model_dir=None):
from faster_whisper import WhisperModel
if model_dir is not None: if model_dir is not None:
print(f"Loading whisper model from model_dir {model_dir}. modelsize and cache_dir parameters are not used.",file=sys.stderr) print(f"Loading whisper model from model_dir {model_dir}. modelsize and cache_dir parameters are not used.",file=sys.stderr)