Update app.py

app.py

@@ -32,14 +32,14 @@ def generate_music_tensors(descriptions, duration: int, device):
     with st.spinner("Generating Music..."):
         # Generate music using the model
         output = model.generate(
-            descriptions=descriptions,
-            progress=True,
-            return_tokens=True,
-            device=device
+        descriptions=descriptions,
+        progress=True,
+        return_tokens=True,
         )
 
-    # Save the generated music audio
-    save_audio(output, device)
+# Save the generated music audio
+# Remove the device argument
+    save_audio(output)
 
     st.success("Music Generation Complete!")
     return output