Update app.py

app.py

@@ -54,20 +54,22 @@ css = """
 with gr.Blocks(css=css) as demo_blocks:
     gr.Markdown(title, elem_id="intro")
 
-    with gr.Row():
+    with gr.Row(elem_id="container"):
         with gr.Column():
             inp_text = gr.Textbox(label="Input Prompt", info="What would you like MusicGen to synthesise?")
-            btn = gr.Button("Generate Music!🎶")
+            btn = gr.Button("Generate Music! 🎶")
             
         with gr.Column():
+            out_vid = gr.make_waveform(animate=True, label=f"Generated Music", show_label=True)
             out = gr.Audio(autoplay=False, label=f"Generated Music", show_label=True,)
 
-
-    with gr.Accordion("Run MusicGen with Transformers 🤗", open=False):
+    with gr.Accordion("Use MusicGen with Transformers 🤗", open=False):
         gr.Markdown(
             """
+            ```python
             import torch
             import soundfile as sf
+            
             from transformers import pipeline
 
             synthesiser = pipeline("text-to-audio", 
@@ -75,14 +77,16 @@ with gr.Blocks(css=css) as demo_blocks:
                                     device="cuda:0", 
                                     torch_dtype=torch.float16)
 
-            music = synthesiser("lo-fi music with a soothing melody", forward_params={"max_new_tokens": 256})
+            music = synthesiser("lo-fi music with a soothing melody", 
+                                forward_params={"max_new_tokens": 256})
 
             sf.write("musicgen_out.wav", music["audio"][0].T, music["sampling_rate"])
+            ```
 
         """
         )
 
-    btn.click(generate_audio, inp_text, out)
+    btn.click(generate_audio, inp_text, [out_vid, out])
     
 
 demo_blocks.queue().launch()