Update app.py

app.py

@@ -7,7 +7,7 @@ import os
 
 # --- AI Model Setup ---
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
-print(f"Device set to use {device}") # Added a print statement for confirmation
+print(f"Device set to use {device}")
 pipe = pipeline("text-to-audio", "facebook/musicgen-small", device=device)
 
 # --- Music Generation Function ---
@@ -35,10 +35,10 @@ iface = gr.Interface(
         gr.Textbox(lines=2, label="Music Description", placeholder="e.g., A futuristic synthwave track..."),
         gr.Slider(minimum=2, maximum=60, value=15, label="Duration (seconds)")
     ],
-    outputs=gr.Audio(type="filepath", label="Generated Music"),
+    outputs=gr.Audio(type="filepath", label="Generated Music", downloadable=True), # Download button enabled here
     title="MusicGen AI",
     description="Transform your ideas into music with AI. Describe a sound and set the duration to get started.",
-    allow_flagging=False # This line disables the feature causing the error
+    allow_flagging="never"
 )
 
 if __name__ == "__main__":