Update app.py

app.py

@@ -40,7 +40,7 @@ def load_model(version):
     return MusicGen.get_pretrained(version)
 
 
-def predict(music_prompt, melody, duration):
+def predict(music_prompt, melody, duration, cfg_coef):
     text = music_prompt
     global MODEL
     topk = int(250)
@@ -54,7 +54,7 @@ def predict(music_prompt, melody, duration):
         top_k=250,
         top_p=0,
         temperature=1.0,
-        cfg_coef=3.0,
+        cfg_coef=cfg_coef,
         duration=duration,
     )
 
@@ -113,6 +113,7 @@ with gr.Blocks(css=css) as demo:
         with gr.Row():
             #model = gr.Radio(["melody", "medium", "small", "large"], label="MusicGen Model", value="melody", interactive=True)
             duration = gr.Slider(minimum=1, maximum=30, value=10, step=1, label="Generated Music Duration", interactive=True)
+            cfg_coef = gr.Number(label="Classifier Free Guidance", minimum=1.0, maximum=10.0, step=0.1, value=3.0, interactive=True)
         with gr.Row():
             submit = gr.Button("Submit")
         #with gr.Row():
@@ -129,34 +130,39 @@ with gr.Blocks(css=css) as demo:
                 [
                     "An 80s driving pop song with heavy drums and synth pads in the background",
                     None,
-                    10
+                    10,
+                    3.0
                 ],
                 [
                     "A cheerful country song with acoustic guitars",
                     None,
-                    10
+                    10,
+                    3.0
                 ],
                 [
                     "90s rock song with electric guitar and heavy drums",
                     None,
-                    10
+                    10,
+                    3.0
                 ],
                 [
                     "a light and cheerly EDM track, with syncopated drums, aery pads, and strong emotions bpm: 130",
                     None,
-                    10
+                    10,
+                    3.0
                 ],
                 [
                     "lofi slow bpm electro chill with organic samples",
                     None,
-                    10
+                    10,
+                    3.0
                 ],
             ],
-            inputs=[music_prompt, melody, duration],
+            inputs=[music_prompt, melody, duration, cfg_coef],
             outputs=[output]
         )
     load_sound_btn.click(split_process, inputs=[uploaded_sound, chosen_track], outputs=[melody])
-    submit.click(predict, inputs=[music_prompt, melody, duration], outputs=[output])
+    submit.click(predict, inputs=[music_prompt, melody, duration, cfg_coef], outputs=[output])
     
 
 demo.queue(max_size=32).launch()