Update controlnet_app.py

controlnet_app.py

@@ -3,6 +3,7 @@ import random
 import numpy as np
 import gradio as gr
 import librosa
+import spaces
 from accelerate import Accelerator
 from transformers import T5Tokenizer, T5EncoderModel
 from diffusers import DDIMScheduler
@@ -70,6 +71,7 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
  tokenizer, text_encoder, noise_scheduler, params) = load_models(config_name, ckpt_path, controlnet_path, vae_path, device)
 
 
+@spaces.GPU
 def generate_audio(text,
                    audio_path, surpass_noise,
                    guidance_scale, guidance_rescale,