Dominik Macháček commited on
Commit
33369a9
·
1 Parent(s): a1ba5e6

import WhisperModel

Browse files
Files changed (1) hide show
  1. whisper_online.py +3 -0
whisper_online.py CHANGED
@@ -6,6 +6,7 @@ from functools import lru_cache
6
  import time
7
  from mosestokenizer import MosesTokenizer
8
 
 
9
  @lru_cache
10
  def load_audio(fname):
11
  a, _ = librosa.load(fname, sr=16000)
@@ -89,6 +90,8 @@ class FasterWhisperASR(ASRBase):
89
  sep = ""
90
 
91
  def load_model(self, modelsize=None, cache_dir=None, model_dir=None):
 
 
92
 
93
  if model_dir is not None:
94
  print(f"Loading whisper model from model_dir {model_dir}. modelsize and cache_dir parameters are not used.",file=sys.stderr)
 
6
  import time
7
  from mosestokenizer import MosesTokenizer
8
 
9
+
10
  @lru_cache
11
  def load_audio(fname):
12
  a, _ = librosa.load(fname, sr=16000)
 
90
  sep = ""
91
 
92
  def load_model(self, modelsize=None, cache_dir=None, model_dir=None):
93
+ from faster_whisper import WhisperModel
94
+
95
 
96
  if model_dir is not None:
97
  print(f"Loading whisper model from model_dir {model_dir}. modelsize and cache_dir parameters are not used.",file=sys.stderr)