Update app.py

app.py

@@ -35,6 +35,8 @@ import gradio as gr
 from stable_audio_tools import get_pretrained_model
 from stable_audio_tools.inference.generation import generate_diffusion_cond
 
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+
 # Load the model outside of the GPU-decorated function
 def load_model():
     print("Loading model...")