Update app.py

app.py

@@ -119,7 +119,7 @@ def generate_music(
 ):
     if use_audio_prompt and not audio_prompt_path:
         raise FileNotFoundError("Please provide an audio prompt filepath when 'use_audio_prompt' is enabled!")
-    max_new_tokens = max_new_tokens * 100  # scaling factor
+    max_new_tokens = max_new_tokens * 50  # scaling factor
 
     with tempfile.TemporaryDirectory() as output_dir:
         stage1_output_dir = os.path.join(output_dir, "stage1")
@@ -326,7 +326,7 @@ def generate_music(
                 return None, None, None
 
 # ------------------ Inference function and Gradio UI ------------------ #
-def infer(genre_txt_content, lyrics_txt_content, num_segments=3, max_new_tokens=5):
+def infer(genre_txt_content, lyrics_txt_content, num_segments=2, max_new_tokens=5):
     try:
         mixed_audio_data, vocal_audio_data, instrumental_audio_data = generate_music(
             genre_txt=genre_txt_content,