Update app.py

app.py

@@ -50,7 +50,7 @@ def generate_script(user_prompt: str, model_id: str, token: str):
 # ---------------------------------------------------------------------
 # Load MusicGen Model (Encapsulated)
 # ---------------------------------------------------------------------
-@spaces.GPU(duration=300)
+
 @spaces.GPU(duration=300)
 def generate_audio(prompt: str, audio_length: int):
     try:
@@ -60,7 +60,7 @@ def generate_audio(prompt: str, audio_length: int):
         musicgen_model.to("cuda")
         inputs = musicgen_processor(text=[prompt], padding=True, return_tensors="pt")
         outputs = musicgen_model.generate(**inputs, max_new_tokens=audio_length)
-        musicgen_model.to("cpu")
+        musicgen_model.to("cpu")  # Return the model to CPU
 
         sr = musicgen_model.config.audio_encoder.sampling_rate
         audio_data = outputs[0, 0].cpu().numpy()
@@ -74,7 +74,6 @@ def generate_audio(prompt: str, audio_length: int):
         return f"Error generating audio: {e}"
 
 
-
 # ---------------------------------------------------------------------
 # Gradio Interface Functions
 # ---------------------------------------------------------------------