Update app.py

app.py

@@ -304,11 +304,11 @@ def stage2_inference(model_stage2, stage1_output_set, stage2_output_dir, batch_s
     return stage2_result
 
 # ----------------------- Main Generation Function (Stage1 + Stage2) -----------------------
-@spaces.GPU(duration=150)
+@spaces.GPU(duration=180)
 def generate_music(
     genre_txt="",
     lyrics_txt="",
-    max_new_tokens=5,
+    max_new_tokens=4,
     run_n_segments=2,
     use_audio_prompt=False,
     audio_prompt_path="",