Update app.py

app.py

@@ -35,17 +35,17 @@ from stable_audio_tools.inference.generation import generate_diffusion_cond
 
 model, model_config = get_pretrained_model("ford442/stable-audio-open-1.0")
 
+device = "cuda" if torch.cuda.is_available() else "cpu"
+print(f"Using device: {device}")
+    
+model.to(device,torch.bfloat16)
+
 # Function to set up, generate, and process the audio
 @spaces.GPU(duration=60)  # Allocate GPU only when this function is called
 def generate_audio(prompt, seconds_total=30, steps=100, cfg_scale=7):
     print(f"Prompt received: {prompt}")
     print(f"Settings: Duration={seconds_total}s, Steps={steps}, CFG Scale={cfg_scale}")
 
-    device = "cuda" if torch.cuda.is_available() else "cpu"
-    print(f"Using device: {device}")
-    
-    model = model.to(device,torch.bfloat16)
-
     # Fetch the Hugging Face token from the environment variable
     hf_token = os.getenv('HF_TOKEN')
     print(f"Hugging Face token: {hf_token}")