Update app.py

app.py

@@ -308,8 +308,8 @@ def stage2_inference(model_stage2, stage1_output_set, stage2_output_dir, batch_s
 def generate_music(
     genre_txt="",
     lyrics_txt="",
-    max_new_tokens=4,
-    run_n_segments=2,
+    max_new_tokens=2,
+    run_n_segments=1,
     use_audio_prompt=False,
     audio_prompt_path="",
     prompt_start_time=0.0,
@@ -555,7 +555,7 @@ with gr.Blocks() as demo:
                 genre_txt = gr.Textbox(label="Genre", placeholder="e.g. Bass Metalcore Thrash Metal Furious bright vocal male")
                 lyrics_txt = gr.Textbox(label="Lyrics", placeholder="Paste lyrics with segments such as [verse], [chorus], etc.")
             with gr.Column():
-                num_segments = gr.Number(label="Number of Segments", value=1, interactive=True)
+                num_segments = gr.Number(label="Number of Segments", value=2, interactive=True)
                 max_new_tokens = gr.Slider(label="Duration of song (sec)", minimum=1, maximum=30, step=1, value=15, interactive=True)
                 use_audio_prompt = gr.Checkbox(label="Use Audio Prompt", value=False)
                 audio_prompt_path = gr.Textbox(label="Audio Prompt Filepath (if used)", placeholder="Path to audio file")