Thao Pham commited on
Commit
9c77f6f
·
1 Parent(s): e9f073f

change whisper model size

Browse files
Files changed (1) hide show
  1. video_utils.py +1 -1
video_utils.py CHANGED
@@ -34,7 +34,7 @@ def extract_audio(path_to_video:str, output_folder:str):
34
  def transcribe_video(path_to_extracted_audio_file, output_folder, whisper_model=None):
35
  # load model
36
  if whisper_model is None:
37
- whisper_model = whisper.load_model("small")
38
  options = dict(task="translate", best_of=1, language='en', verbose=True)
39
  results = whisper_model.transcribe(path_to_extracted_audio_file, **options)
40
 
 
34
  def transcribe_video(path_to_extracted_audio_file, output_folder, whisper_model=None):
35
  # load model
36
  if whisper_model is None:
37
+ whisper_model = whisper.load_model("tiny")
38
  options = dict(task="translate", best_of=1, language='en', verbose=True)
39
  results = whisper_model.transcribe(path_to_extracted_audio_file, **options)
40