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
|