marquesafonso commited on
Commit
51dc230
·
1 Parent(s): 555839a

roll back to large-v3

Browse files
Files changed (1) hide show
  1. src/transcriber.py +2 -2
src/transcriber.py CHANGED
@@ -48,8 +48,8 @@ def transcriber(video_input:gr.File,
48
  task:str):
49
  srt_filepath = os.path.normpath(f"{video_input.split('.')[0]}.srt")
50
  audio_input = convert_video_to_audio(video_input)
51
- model_size = "distil-large-v3"
52
- model = WhisperModel(model_size, device="cpu", compute_type="int8")
53
  segments, _ = model.transcribe(
54
  audio_input,
55
  beam_size=5,
 
48
  task:str):
49
  srt_filepath = os.path.normpath(f"{video_input.split('.')[0]}.srt")
50
  audio_input = convert_video_to_audio(video_input)
51
+ model_version = "large-v3"
52
+ model = WhisperModel(model_version, device="cpu", compute_type="int8")
53
  segments, _ = model.transcribe(
54
  audio_input,
55
  beam_size=5,