Spaces:
Paused
Paused
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
|