sdafd commited on
Commit
a62b699
·
verified ·
1 Parent(s): 77c2d4d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -21,6 +21,7 @@ models = {
21
  "base": whisperx.load_model("base", device, compute_type=compute_type, vad_method='silero'),
22
  "small": whisperx.load_model("small", device, compute_type=compute_type, vad_method='silero'),
23
  "large": whisperx.load_model("large", device, compute_type=compute_type, vad_method='silero'),
 
24
  }
25
 
26
  def transcribe(audio_file, model_size="base", debug=False):
 
21
  "base": whisperx.load_model("base", device, compute_type=compute_type, vad_method='silero'),
22
  "small": whisperx.load_model("small", device, compute_type=compute_type, vad_method='silero'),
23
  "large": whisperx.load_model("large", device, compute_type=compute_type, vad_method='silero'),
24
+ "large-v2": whisperx.load_model("large-v2", device, compute_type=compute_type, vad_method='silero'),
25
  }
26
 
27
  def transcribe(audio_file, model_size="base", debug=False):