Update app.py

app.py

@@ -35,7 +35,7 @@ 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")
+device = torch.device("cuda:0")
 
 # Load the model outside of the GPU-decorated function
 def load_model():