Update app.py

app.py

@@ -123,7 +123,7 @@ def generate_music(
         genre_txt=None,
         lyrics_txt=None,
         run_n_segments=2,
-        max_new_tokens=5,
+        max_new_tokens=22,
         use_audio_prompt=False,
         audio_prompt_path="",
         prompt_start_time=0.0,
@@ -220,7 +220,7 @@ def generate_music(
                     logits_processor=LogitsProcessorList([BlockTokenRangeProcessor(0, 32002), BlockTokenRangeProcessor(32016, 32016)]),
                     guidance_scale=guidance_scale,
                     use_cache=True,
-                    num_beams=2
+                    num_beams=3
                 )
                 if output_seq[0][-1].item() != mmtokenizer.eoa:
                     tensor_eoa = torch.as_tensor([[mmtokenizer.eoa]]).to(model.device)