devingulliver commited on
Commit
d818586
·
verified ·
1 Parent(s): 5b9537a

Revert recent changes

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,12 +14,12 @@ urllib.request.urlretrieve(
14
 
15
  writer = whisper.utils.get_writer("srt", "/dev/null")
16
 
17
- @spaces.GPU#(duration=120)
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, task="translate")
23
  out = io.StringIO()
24
  writer.write_result(result, out)
25
  return out.getvalue()
 
14
 
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()