Thao Pham
commited on
Commit
·
9c77f6f
1
Parent(s):
e9f073f
change whisper model size
Browse files- 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("
|
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 |
|