Update app.py

app.py

@@ -40,12 +40,12 @@ def load_model(version):
     return MusicGen.get_pretrained(version)
 
 
-def predict(music_prompt, melody, duration, model):
+def predict(music_prompt, melody, duration):
     text = music_prompt
     global MODEL
     topk = int(250)
-    if MODEL is None or MODEL.name != model:
-        MODEL = load_model(model)
+    if MODEL is None or MODEL.name != "melody":
+        MODEL = load_model("melody")
 
     if duration > MODEL.lm.cfg.dataset.segment_duration:
         raise gr.Error("MusicGen currently supports durations of up to 30 seconds!")
@@ -111,7 +111,7 @@ with gr.Blocks(css=css) as demo:
             music_prompt = gr.Textbox(label="Musical Prompt", info="Describe what kind of music you wish for", interactive=True, placeholder="lofi slow bpm electro chill with organic samples")
             melody = gr.Audio(source="upload", type="numpy", label="Track Condition (from previous step)", interactive=False)
         with gr.Row():
-            model = gr.Radio(["melody", "medium", "small", "large"], label="MusicGen Model", value="melody", interactive=True)
+            #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)
         with gr.Row():
             submit = gr.Button("Submit")
@@ -129,37 +129,34 @@ with gr.Blocks(css=css) as demo:
                 [
                     "An 80s driving pop song with heavy drums and synth pads in the background",
                     None,
-                    10,
-                    "melody"
+                    10
                 ],
                 [
                     "A cheerful country song with acoustic guitars",
                     None,
-                    10,
-                    "large"
+                    10
                 ],
                 [
                     "90s rock song with electric guitar and heavy drums",
                     None,
+                    10
                 ],
                 [
                     "a light and cheerly EDM track, with syncopated drums, aery pads, and strong emotions bpm: 130",
                     None,
-                    10,
-                    "melody"
+                    10
                 ],
                 [
                     "lofi slow bpm electro chill with organic samples",
                     None,
-                    10,
-                    "small"
+                    10
                 ],
             ],
-            inputs=[music_prompt, melody, duration, model],
+            inputs=[music_prompt, melody, duration],
             outputs=[output]
         )
     load_sound_btn.click(split_process, inputs=[uploaded_sound, chosen_track], outputs=[melody])
-    submit.click(predict, inputs=[music_prompt, melody, duration, model], outputs=[output])
+    submit.click(predict, inputs=[music_prompt, melody, duration], outputs=[output])
     
 
 demo.queue(max_size=32).launch()