Update app.py

app.py

@@ -38,12 +38,12 @@ def load_model(version):
     return MusicGen.get_pretrained(version)
 
 
-def predict(music_prompt, melody, duration):
+def predict(music_prompt, melody, duration, model):
     text = music_prompt
     global MODEL
     topk = int(250)
-    if MODEL is None or MODEL.name != "melody":
-        MODEL = load_model("melody")
+    if MODEL is None or MODEL.name != model:
+        MODEL = load_model(model)
 
     if duration > MODEL.lm.cfg.dataset.segment_duration:
         raise gr.Error("MusicGen currently supports durations of up to 30 seconds!")
@@ -104,10 +104,10 @@ with gr.Blocks(css=css) as demo:
             #split_drums = gr.Audio(type="filepath", label="Drums")
             #split_others = gr.Audio(type="filepath", label="Other")
         with gr.Row():
-            music_prompt = gr.Textbox(label="Musical Prompt", info="Describe what kind music you wish for", interactive=True)
-            melody = gr.Audio(source="upload", type="numpy", label="Melody Condition (from previous step)", interactive=False)
-        #with gr.Row():
-        #    model = gr.Radio(["melody", "medium", "small", "large"], label="Model", value="melody", interactive=True)
+            music_prompt = gr.Textbox(label="Musical Prompt", info="Describe what kind of music you wish for", interactive=True)
+            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="Model", value="melody", interactive=True)
         with gr.Row():
             duration = gr.Slider(minimum=1, maximum=30, value=10, step=1, label="Duration", interactive=True)
         with gr.Row():
@@ -119,7 +119,7 @@ with gr.Blocks(css=css) as demo:
         #   cfg_coef = gr.Number(label="Classifier Free Guidance", value=3.0, interactive=True)
         
         output = gr.Audio(label="Generated Music")
-    load_sound_btn.click(split_process, inputs=[uploaded_sound, chosen_track], outputs=[melody])
+    load_sound_btn.click(split_process, inputs=[uploaded_sound, chosen_track, model], outputs=[melody])
     submit.click(predict, inputs=[music_prompt, melody, duration], outputs=[output])