Prev was by 03 and i solved error by him in this one

app.py

@@ -276,10 +276,10 @@ def stage2_inference(model_stage2, stage1_output_set, stage2_output_dir, batch_s
 # ----------------------- Main Generation Function (Stage1 + Stage2) -----------------------
 @spaces.GPU(duration=120)
 def generate_music(
-    max_new_tokens=5,
-    run_n_segments=2,
     genre_txt="",
     lyrics_txt="",
+    max_new_tokens=5,
+    run_n_segments=2,
     use_audio_prompt=False,
     audio_prompt_path="",
     prompt_start_time=0.0,
@@ -590,7 +590,7 @@ Living out my dreams with this mic and a deal
         )
     submit_btn.click(
         fn=generate_music,
-        inputs=[max_new_tokens, num_segments, genre_txt, lyrics_txt, use_audio_prompt, audio_prompt_path],
+        inputs=[genre_txt, lyrics_txt, max_new_tokens, num_segments, use_audio_prompt, audio_prompt_path],
         outputs=[music_out, vocal_out, instrumental_out]
     )
     gr.Markdown("## Contributions Welcome\nFeel free to contribute improvements or fixes.")