katospiegel commited on
Commit
f3782dc
·
1 Parent(s): da53eff

faster whisper timestamp

Browse files
Files changed (1) hide show
  1. transcription.py +1 -1
transcription.py CHANGED
@@ -61,7 +61,7 @@ def fast_transcription(audio_file, whisper_model, language):
61
  print(f"conversion to wav ready, duration of audio file: {duration}")
62
 
63
  # Transcribe audio
64
- options = dict(language=language, beam_size=5, best_of=5)
65
  transcribe_options = dict(task="transcribe", **options)
66
  segments_generator, info = model.transcribe(audio_file, **transcribe_options)
67
 
 
61
  print(f"conversion to wav ready, duration of audio file: {duration}")
62
 
63
  # Transcribe audio
64
+ options = dict(language=language, beam_size=5, best_of=5, word_timestamps=True)
65
  transcribe_options = dict(task="transcribe", **options)
66
  segments_generator, info = model.transcribe(audio_file, **transcribe_options)
67