Spaces:
Sleeping
Sleeping
Michael Hu
commited on
Commit
·
6f92dbc
1
Parent(s):
6034fea
use unsloth whisper model to reduce the model size
Browse files- utils/stt.py +1 -1
utils/stt.py
CHANGED
|
@@ -62,7 +62,7 @@ class WhisperModel(ASRModel):
|
|
| 62 |
use_safetensors=True
|
| 63 |
).to(self.device)
|
| 64 |
|
| 65 |
-
self.processor = AutoProcessor.from_pretrained("
|
| 66 |
logger.info("Whisper model loaded successfully")
|
| 67 |
|
| 68 |
def transcribe(self, audio_path):
|
|
|
|
| 62 |
use_safetensors=True
|
| 63 |
).to(self.device)
|
| 64 |
|
| 65 |
+
self.processor = AutoProcessor.from_pretrained("unsloth/whisper-large-v3")
|
| 66 |
logger.info("Whisper model loaded successfully")
|
| 67 |
|
| 68 |
def transcribe(self, audio_path):
|