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=22,
+        max_new_tokens=35,
         use_audio_prompt=False,
         audio_prompt_path="",
         prompt_start_time=0.0,
@@ -210,7 +210,7 @@ def generate_music(
                 output_seq = model.generate(
                     input_ids=input_ids,
                     max_new_tokens=max_new_tokens,
-                    min_new_tokens=100,  # Keep min_new_tokens to avoid short generations
+                    min_new_tokens=100,
                     do_sample=True,
                     top_p=top_p,
                     temperature=temperature,
@@ -230,7 +230,6 @@ def generate_music(
                 raw_output = torch.cat([raw_output, prompt_ids, output_seq[:, input_ids.shape[-1]:]], dim=1)
             else:
                 raw_output = output_seq
-            print(len(raw_output))
 
         # save raw output and check sanity
         ids = raw_output[0].cpu().numpy()