yellowcandle commited on
Commit
bca5261
·
unverified ·
1 Parent(s): a28c209
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ from transformers import AutoModelForSpeechSeq2Seq, AutoProcessor, pipeline, Aut
8
  transcribe_model = None
9
  proofread_model = None
10
 
11
- @spaces.gpu(duration=60)
12
  def transcribe_audio(audio):
13
  global transcribe_model
14
  if audio is None:
@@ -36,7 +36,7 @@ def transcribe_audio(audio):
36
  result = pipe(audio)
37
  return result["text"]
38
 
39
- @spaces.gpu(duration=120)
40
  def proofread(text):
41
  global proofread_model
42
  if text is None:
 
8
  transcribe_model = None
9
  proofread_model = None
10
 
11
+ @spaces.GPU(duration=60)
12
  def transcribe_audio(audio):
13
  global transcribe_model
14
  if audio is None:
 
36
  result = pipe(audio)
37
  return result["text"]
38
 
39
+ @spaces.GPU(duration=120)
40
  def proofread(text):
41
  global proofread_model
42
  if text is None: