Update app.py

app.py

@@ -275,6 +275,8 @@ def generate_music(prime, num_gen_tokens, num_mem_tokens, num_gen_batches, model
         out = model.generate(
             inp,
             num_gen_tokens,
+            filter_logits_fn=top_p,
+            filter_kwargs={'thres': 0.96},
             temperature=model_temperature,
             return_prime=False,
             verbose=False