devingulliver commited on
Commit
44800c6
1 Parent(s): e846051

switch back to distil large v3

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,11 +5,11 @@ import io
5
  import whisper
6
  from huggingface_hub import hf_hub_download
7
 
8
- model_path = hf_hub_download(repo_id="distil-whisper/distil-medium.en", filename="original-model.bin")
9
 
10
  writer = whisper.utils.get_writer("srt", "/dev/null")
11
 
12
- @spaces.GPU
13
  def generate(file, progress=gr.Progress(track_tqdm=True)):
14
  # get file to type bytes somehow
15
  model = whisper.load_model(model_path, device="cuda")
 
5
  import whisper
6
  from huggingface_hub import hf_hub_download
7
 
8
+ model_path = hf_hub_download(repo_id="distil-whisper/distil-large-v3-openai", filename="model.bin")
9
 
10
  writer = whisper.utils.get_writer("srt", "/dev/null")
11
 
12
+ @spaces.GPU(duration=90)
13
  def generate(file, progress=gr.Progress(track_tqdm=True)):
14
  # get file to type bytes somehow
15
  model = whisper.load_model(model_path, device="cuda")