Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import spaces
 import huggingface_hub
 import numpy as np
 import pandas as pd
@@ -64,6 +65,7 @@ def check_for_wav_in_outputs():
     else:
         return None
 
+@spaces.GPU()
 def infer(prompt_in, chords, duration, bpms):
 
     # check if 'outputs' dir exists and empty it if necessary
@@ -91,11 +93,12 @@ def infer(prompt_in, chords, duration, bpms):
     wav = []
     for i in range(num_samples//bs):
         print(f"starting {i} batch...")
-        temp = musicgen.generate_with_chords_and_beats(descriptions[i*bs:(i+1)*bs], 
-                                                          chords[i*bs:(i+1)*bs],
-                                                          bpms[i*bs:(i+1)*bs], 
-                                                          meters[i*bs:(i+1)*bs]
-                                                          )
+        temp = musicgen.generate_with_chords_and_beats(
+            descriptions[i*bs:(i+1)*bs],
+            chords[i*bs:(i+1)*bs],
+            bpms[i*bs:(i+1)*bs],
+            meters[i*bs:(i+1)*bs]
+        )
         wav.extend(temp.cpu())
     
     # save and display generated audio
@@ -114,6 +117,9 @@ css="""
     max-width: 800px;
     margin: 0 auto;
 }
+#chords-examples button{
+    font-size: 20px;
+}
 """
 
 with gr.Blocks(css=css) as demo:
@@ -123,7 +129,7 @@ with gr.Blocks(css=css) as demo:
         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)
+            duration = gr.Slider(label="Sample duration", minimum=4, maximum=30, step=1, value=30)
             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")
@@ -141,6 +147,7 @@ with gr.Blocks(css=css) as demo:
                 )
                 gr.Examples(
                     label = "Chords progression examples",
+                    elem_id = "chords-examples",
                     examples = ['C G A:min F',
                                 'A:min F C G',
                                 'C F G F',