Luigi commited on
Commit
c9a0c53
·
1 Parent(s): 2899031

Add more fwhisper models

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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()