AnyaSchen commited on
Commit
d3fe52c
·
1 Parent(s): b67cb1c

fix: print using model

Browse files
whisper_streaming_custom/backends.py CHANGED
@@ -48,6 +48,7 @@ class WhisperTimestampedASR(ASRBase):
48
  sep = " "
49
 
50
  def load_model(self, modelsize=None, cache_dir=None, model_dir=None):
 
51
  import whisper
52
  import whisper_timestamped
53
  from whisper_timestamped import transcribe_timestamped
@@ -128,6 +129,7 @@ class FasterWhisperASR(ASRBase):
128
  sep = ""
129
 
130
  def load_model(self, modelsize=None, cache_dir=None, model_dir=None):
 
131
  from faster_whisper import WhisperModel
132
 
133
  if model_dir is not None:
@@ -230,6 +232,7 @@ class MLXWhisper(ASRBase):
230
  sep = ""
231
 
232
  def load_model(self, modelsize=None, cache_dir=None, model_dir=None):
 
233
  from mlx_whisper.transcribe import ModelHolder, transcribe
234
  import mlx.core as mx
235
 
@@ -308,6 +311,7 @@ class MLXWhisper(ASRBase):
308
  class OpenaiApiASR(ASRBase):
309
  """Uses OpenAI's Whisper API for transcription."""
310
  def __init__(self, lan=None, temperature=0, logfile=sys.stderr):
 
311
  self.logfile = logfile
312
  self.modelname = "whisper-1"
313
  self.original_language = None if lan == "auto" else lan
 
48
  sep = " "
49
 
50
  def load_model(self, modelsize=None, cache_dir=None, model_dir=None):
51
+ print("Loading whisper_timestamped model")
52
  import whisper
53
  import whisper_timestamped
54
  from whisper_timestamped import transcribe_timestamped
 
129
  sep = ""
130
 
131
  def load_model(self, modelsize=None, cache_dir=None, model_dir=None):
132
+ print("Loading faster-whisper model")
133
  from faster_whisper import WhisperModel
134
 
135
  if model_dir is not None:
 
232
  sep = ""
233
 
234
  def load_model(self, modelsize=None, cache_dir=None, model_dir=None):
235
+ print("Loading mlx whisper model")
236
  from mlx_whisper.transcribe import ModelHolder, transcribe
237
  import mlx.core as mx
238
 
 
311
  class OpenaiApiASR(ASRBase):
312
  """Uses OpenAI's Whisper API for transcription."""
313
  def __init__(self, lan=None, temperature=0, logfile=sys.stderr):
314
+ print("Loading openai api model")
315
  self.logfile = logfile
316
  self.modelname = "whisper-1"
317
  self.original_language = None if lan == "auto" else lan