Helw150 commited on
Commit
4898006
1 Parent(s): 0e3a570

Spaces GPU Move?

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -21,7 +21,7 @@ if gr.NO_RELOAD:
21
  resampler = Audio(sampling_rate=16_000)
22
 
23
 
24
- @spaces.GPU(duration=20, progress=gr.Progress(track_tqdm=True))
25
  @torch.no_grad
26
  def diva_audio(audio_input, do_sample=False, temperature=0.001, prev_outs=None):
27
  sr, y = audio_input
@@ -58,6 +58,7 @@ def process_audio(audio: tuple, state: AppState):
58
  return audio, state
59
 
60
 
 
61
  def response(state: AppState, audio: tuple):
62
  if not audio:
63
  return AppState()
 
21
  resampler = Audio(sampling_rate=16_000)
22
 
23
 
24
+ @spaces.GPU
25
  @torch.no_grad
26
  def diva_audio(audio_input, do_sample=False, temperature=0.001, prev_outs=None):
27
  sr, y = audio_input
 
58
  return audio, state
59
 
60
 
61
+ @spaces.GPU(duration=40, progress=gr.Progress(track_tqdm=True))
62
  def response(state: AppState, audio: tuple):
63
  if not audio:
64
  return AppState()