devingulliver commited on
Commit
19001c8
·
verified ·
1 Parent(s): d818586
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,11 +15,11 @@ urllib.request.urlretrieve(
15
  writer = whisper.utils.get_writer("srt", "/dev/null")
16
 
17
  @spaces.GPU
18
- def generate(file):
19
  # get file to type bytes somehow
20
  model = whisper.load_model("large", device="cuda")
21
  audio = whisper.load_audio(file)
22
- result = model.transcribe(audio)
23
  out = io.StringIO()
24
  writer.write_result(result, out)
25
  return out.getvalue()
 
15
  writer = whisper.utils.get_writer("srt", "/dev/null")
16
 
17
  @spaces.GPU
18
+ def generate(file, progress=gr.Progress(track_tqdm=True)):
19
  # get file to type bytes somehow
20
  model = whisper.load_model("large", device="cuda")
21
  audio = whisper.load_audio(file)
22
+ result = model.transcribe(audio, verbose=False)
23
  out = io.StringIO()
24
  writer.write_result(result, out)
25
  return out.getvalue()