Update app.py

app.py

@@ -10,13 +10,20 @@ from huggingface_hub import login
 hf_token = os.getenv('HF_TOKEN')
 login(token=hf_token,add_to_git_credential=True)
 
+# Load the model outside of the GPU-decorated function
+def load_model():
+    print("Loading model...")
+    model, model_config = get_pretrained_model("stabilityai/stable-audio-open-1.0")
+    print("Model loaded successfully.")
+    return model, model_config
+
 # Define the function to generate audio
 @spaces.GPU(duration=120)
 def generate_audio(prompt, bpm, seconds_total):
     device = "cuda" if torch.cuda.is_available() else "cpu"
 
     # Download model
-    model, model_config = get_pretrained_model("stabilityai/stable-audio-open-1.0")
+    model, model_config = load_model
     sample_rate = model_config["sample_rate"]
     sample_size = model_config["sample_size"]