Update app.py

app.py

@@ -160,7 +160,7 @@ def generate_music(
         # Decoding config
         top_p = 0.93
         temperature = 1.0
-        repetition_penalty = 1.3
+        repetition_penalty = 1.2
         start_of_segment = mmtokenizer.tokenize('[start_of_segment]')
         end_of_segment = mmtokenizer.tokenize('[end_of_segment]')
 
@@ -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=5
                 )
                 if output_seq[0][-1].item() != mmtokenizer.eoa:
                     tensor_eoa = torch.as_tensor([[mmtokenizer.eoa]]).to(model.device)