Complete the examples

demos/musicgen_app.py

@@ -264,10 +264,6 @@ def ui_full(launch_kwargs):
                                           interactive=True, elem_id="melody-input")
                 with gr.Row():
                     duration = gr.Slider(label = "Duration", info = "(in seconds)", minimum = 1, maximum = 120, value = 10, interactive = True)
-                with gr.Row():
-                    submit = gr.Button("Generate", variant = "primary")
-                    # Adapted from https://github.com/rkfg/audiocraft/blob/long/app.py, MIT license.
-                    _ = gr.Button("Abort all", variant="stop").click(fn=interrupt, queue = False)
                 with gr.Accordion("Advanced options", open = False):
                     with gr.Row():
                         topk = gr.Number(label="Top-k", info = "Number of tokens shortlisted", value = 250, interactive = True)
@@ -286,9 +282,13 @@ def ui_full(launch_kwargs):
                                           "facebook/musicgen-stereo-melody-large"],
                                          label="Model", value="facebook/musicgen-stereo-melody", interactive=True)
                         model_path = gr.Text(label="Model Path (custom models)")
+                with gr.Row():
+                    submit = gr.Button("Generate", variant = "primary")
+                    # Adapted from https://github.com/rkfg/audiocraft/blob/long/app.py, MIT license.
+                    _ = gr.Button("Abort all", variant="stop").click(fn=interrupt, queue = False)
             with gr.Column():
-                output = gr.Video(label="Generated Music")
-                audio_output = gr.Audio(label="Generated Music (wav)", type='filepath')
+                output = gr.Video(label="Generated Music", autoplay = True)
+                audio_output = gr.Audio(label="Generated Music (wav)", type='filepath', show_download_button = True)
                 diffusion_output = gr.Video(label="MultiBand Diffusion Decoder")
                 audio_diffusion = gr.Audio(label="MultiBand Diffusion Decoder (wav)", type='filepath')
         submit.click(toggle_diffusion, decoder, [diffusion_output, audio_diffusion], queue=False,
@@ -303,17 +303,19 @@ def ui_full(launch_kwargs):
                 [
                     "An angry propulsive industrial score with distorted synthesizers and tortured vocals.",
                     None,
+                    10,
                     "facebook/musicgen-stereo-melody",
                     "Default"
                 ],
                 [
                     "A monstrous industrial bach hybrid",
                     "./assets/bach.mp3",
+                    10,
                     "facebook/musicgen-stereo-melody",
                     "Default"
                 ],
             ],
-            inputs=[text, melody, model, decoder],
+            inputs=[text, melody, duration, model, decoder],
             outputs=[output]
         )
         gr.Markdown(