Nithya commited on
Commit
f7854be
·
1 Parent(s): f2917d8

added spaces GPU decorator

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  from gradio import Interface, Audio
2
  import gradio as gr
3
  import numpy as np
@@ -162,7 +163,8 @@ def generate_audio(audio_model, f0s, invert_audio_fn, outfolder, singers=[3], nu
162
  logging.log(logging.INFO, f"Saving audio {i}")
163
  torchaudio.save(f"{outfolder}/{i}.wav", torch.tensor(a).detach().unsqueeze(0).cpu(), 16000)
164
  return audio
165
-
 
166
  def generate(pitch, num_samples=2, num_steps=100, singers=[3], outfolder='temp', audio_seq_len=750, pitch_qt=None ):
167
  global global_ind, audio_components
168
  global preprocessed_primes
 
1
+ import spaces
2
  from gradio import Interface, Audio
3
  import gradio as gr
4
  import numpy as np
 
163
  logging.log(logging.INFO, f"Saving audio {i}")
164
  torchaudio.save(f"{outfolder}/{i}.wav", torch.tensor(a).detach().unsqueeze(0).cpu(), 16000)
165
  return audio
166
+
167
+ @spaces.GPU(duration=180)
168
  def generate(pitch, num_samples=2, num_steps=100, singers=[3], outfolder='temp', audio_seq_len=750, pitch_qt=None ):
169
  global global_ind, audio_components
170
  global preprocessed_primes