Spaces:
Running
on
Zero
Running
on
Zero
Add more fwhisper models
Browse files
app.py
CHANGED
@@ -193,6 +193,7 @@ def _transcribe_whisper_cpu(model_id, language, audio_path, enable_diar):
|
|
193 |
@spaces.GPU
|
194 |
def _transcribe_whisper_gpu(model_id, language, audio_path, enable_diar):
|
195 |
pipe = get_fwhisper_model(model_id, "cuda")
|
|
|
196 |
# Diarization-only branch
|
197 |
if enable_diar:
|
198 |
diarizer = get_diarization_pipe()
|
|
|
193 |
@spaces.GPU
|
194 |
def _transcribe_whisper_gpu(model_id, language, audio_path, enable_diar):
|
195 |
pipe = get_fwhisper_model(model_id, "cuda")
|
196 |
+
cprint('Whisper (faster-whisper) using CUDA', 'green')
|
197 |
# Diarization-only branch
|
198 |
if enable_diar:
|
199 |
diarizer = get_diarization_pipe()
|