Update app.py

app.py

@@ -119,8 +119,8 @@ 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, model], outputs=[melody])
-    submit.click(predict, 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])
     
 
 demo.queue(max_size=32).launch()