Update app.py

app.py

@@ -80,10 +80,8 @@ def infer(prompt_in, chords, duration, bpms):
     musicgen = audiocraft.models.MusicGen.get_pretrained('./ckpt/musicongen') ### change this path
     musicgen.set_generation_params(duration=duration, extend_stride=duration//2, top_k = 250)
     
-    
-    # chords = [chords]
-    chords = ['C G A:min F', 'A:min F C G', 'C F G F']
-    
+    chords = [chords]
+     
     descriptions = [prompt_in] * num_samples
     
     bpms = [bpms] * num_samples
@@ -121,26 +119,36 @@ css="""
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown("#MusiConGen")
-        with gr.Row():
-            with gr.Column():
-                prompt_in = gr.Textbox(label="Music description", value="A laid-back blues shuffle with a relaxed tempo, warm guitar tones, and a comfortable groove, perfect for a slow dance or a night in. Instruments: electric guitar, bass, drums.")
-                chords = gr.Textbox(label="Chords progression", value='C G A:min F')
-                duration = gr.Slider(label="Sample duration", minimum=4, maximum=30, step=1, value=20)
-                bpms = gr.Slider(label="BPMs", minimum=50, maximum=220, step=1, value=120)
-                submit_btn = gr.Button("Submit")
-            wav_out = gr.Audio(label="Wav Result")
-        gr.Examples(
-            label = "Audio description examples",
-            examples = [
-                ["A laid-back blues shuffle with a relaxed tempo, warm guitar tones, and a comfortable groove, perfect for a slow dance or a night in. Instruments: electric guitar, bass, drums."],
-                ["A smooth acid jazz track with a laid-back groove, silky electric piano, and a cool bass, providing a modern take on jazz. Instruments: electric piano, bass, drums."],
-                ["A classic rock n' roll tune with catchy guitar riffs, driving drums, and a pulsating bass line, reminiscent of the golden era of rock. Instruments: electric guitar, bass, drums."],
-                ["A high-energy funk tune with slap bass, rhythmic guitar riffs, and a tight horn section, guaranteed to get you grooving. Instruments: bass, guitar, trumpet, saxophone, drums."],
-                ["A heavy metal onslaught with double kick drum madness, aggressive guitar riffs, and an unrelenting bass, embodying the spirit of metal. Instruments: electric guitar, bass guitar, drums."]
-            ],
-            inputs = [prompt_in]
-        )
         
+        with gr.Column():
+            prompt_in = gr.Textbox(label="Music description", value="A laid-back blues shuffle with a relaxed tempo, warm guitar tones, and a comfortable groove, perfect for a slow dance or a night in. Instruments: electric guitar, bass, drums.")
+            chords = gr.Textbox(label="Chords progression", value='C G A:min F')
+            duration = gr.Slider(label="Sample duration", minimum=4, maximum=30, step=1, value=20)
+            bpms = gr.Slider(label="BPMs", minimum=50, maximum=220, step=1, value=120)
+            submit_btn = gr.Button("Submit")
+            wav_out = gr.Audio(label="Wav Result")
+            with gr.Row():
+                gr.Examples(
+                    label = "Audio description examples",
+                    examples = [
+                        ["A laid-back blues shuffle with a relaxed tempo, warm guitar tones, and a comfortable groove, perfect for a slow dance or a night in. Instruments: electric guitar, bass, drums."],
+                        ["A smooth acid jazz track with a laid-back groove, silky electric piano, and a cool bass, providing a modern take on jazz. Instruments: electric piano, bass, drums."],
+                        ["A classic rock n' roll tune with catchy guitar riffs, driving drums, and a pulsating bass line, reminiscent of the golden era of rock. Instruments: electric guitar, bass, drums."],
+                        ["A high-energy funk tune with slap bass, rhythmic guitar riffs, and a tight horn section, guaranteed to get you grooving. Instruments: bass, guitar, trumpet, saxophone, drums."],
+                        ["A heavy metal onslaught with double kick drum madness, aggressive guitar riffs, and an unrelenting bass, embodying the spirit of metal. Instruments: electric guitar, bass guitar, drums."]
+                    ],
+                    inputs = [prompt_in]
+                )
+                gr.Examples(
+                    label = "Chords progression examples",
+                    examples = ['C G A:min F',
+                                'A:min F C G',
+                                'C F G F',
+                                'C A:min F G',
+                                'D:min G C A:min'
+                               ],
+                    inputs = [chords]
+                )
         
     submit_btn.click(
         fn = infer,