Update main.py

main.py

@@ -1,3 +1,4 @@
+import spaces 
 import gradio as gr
 import torch
 import gc
@@ -11,7 +12,6 @@ from audiosr import build_model, super_resolution
 from scipy import signal
 import pyloudnorm as pyln
 import tempfile
-import spaces 
 
 
 class AudioUpscaler:
@@ -363,7 +363,6 @@ class AudioUpscaler:
         return waveform
         # return output_file
 
-@spaces.GPU
 def inference(audio_file, model_name, guidance_scale, ddim_steps, seed):
     audiosr = build_model(model_name=model_name)
 
@@ -386,6 +385,7 @@ def inference(audio_file, model_name, guidance_scale, ddim_steps, seed):
     
     return (48000, waveform)
 
+@spaces.GPU
 def upscale_audio(
     input_file,
     output_folder,