Update app.py

app.py

@@ -149,10 +149,10 @@ def model_inference(input_ids, max_new_tokens, top_p, temperature, repetition_pe
 
 
 def generate_music(
-        max_new_tokens=5,
-        run_n_segments=2,
         genre_txt=None,
         lyrics_txt=None,
+        run_n_segments=2,
+        max_new_tokens=15,
         use_audio_prompt=False,
         audio_prompt_path="",
         prompt_start_time=0.0,
@@ -417,11 +417,11 @@ Locked inside my mind, hot flame.
             outputs=[music_out, vocal_out, instrumental_out],
             cache_examples=True,
             cache_mode="eager",
-            fn=infer
+            fn=generate_music
         )
 
     submit_btn.click(
-        fn=infer,
+        fn=generate_music,
         inputs=[genre_txt, lyrics_txt, num_segments, max_new_tokens],
         outputs=[music_out, vocal_out, instrumental_out]
     )