Update app.py

app.py

@@ -291,7 +291,7 @@ def generate_music(prime, num_gen_tokens, num_mem_tokens, num_gen_batches, model
     return out.tolist()
 
 def generate_music_and_state(input_midi, num_prime_tokens, num_gen_tokens, num_mem_tokens,
-                             model_temperature, final_composition, generated_batches, block_lines):
+                             model_temperature, model_top_p, add_drums, add_outro, final_composition, generated_batches, block_lines):
     """
     Generate tokens using the model, update the composition state, and prepare outputs.
     This function combines seed loading, token generation, and UI output packaging.
@@ -450,6 +450,9 @@ with gr.Blocks() as demo:
     num_gen_tokens = gr.Slider(16, 1024, value=512, step=1, label="Number of tokens to generate")
     num_mem_tokens = gr.Slider(16, 8192, value=8192, step=1, label="Number of memory tokens")
     model_temperature = gr.Slider(0.1, 1, value=0.9, step=0.01, label="Model temperature")
+    model_top_p = gr.Slider(0.1, 0.99, value=0.96, step=0.01, label="Model sampling top p value")
+    add_drums = gr.Checkbox(value="False", label="Add drums")
+    add_outro = gr.Checkbox(value="False", label="Add an outro")
     generate_btn = gr.Button("Generate", variant="primary")
 
     gr.Markdown("## Batch Previews")
@@ -462,7 +465,7 @@ with gr.Blocks() as demo:
             outputs.extend([audio_output, plot_output])
     generate_btn.click(
         generate_music_and_state,
-        [input_midi, num_prime_tokens, num_gen_tokens, num_mem_tokens, model_temperature,
+        [input_midi, num_prime_tokens, num_gen_tokens, num_mem_tokens, model_temperature, model_top_p, add_drums, add_outro,
          final_composition, generated_batches, block_lines],
         outputs
     )