Update app.py

app.py

@@ -186,10 +186,10 @@ def load_midi(input_midi):
 #==================================================================================
 
 @spaces.GPU
-def Generate_Drums(input_midi,
-                   model_temperature,
-                   model_sampling_top_p
-                  ):
+def Humanize_MIDI(input_midi,
+                  model_temperature,
+                  model_sampling_top_p
+                 ):
 
     #===============================================================================
 
@@ -428,6 +428,11 @@ with gr.Blocks() as demo:
     
     gr.Markdown("## Generation options")
 
+    num_prime_toks = gr.Slider(0, 1024, value=16, step=1, label="Number of prime tokens")
+
+    humanize_durations = gr.Checkbox(value=False, label="Humanize durations")
+    humanize_velocities = gr.Checkbox(value=True, label="Humanize velocities")
+
     model_temperature = gr.Slider(0.1, 1, value=0.9, step=0.01, label="Model temperature")
     model_sampling_top_p = gr.Slider(0.1, 0.99, value=0.96, step=0.01, label="Model sampling top p value")
     
@@ -440,7 +445,7 @@ with gr.Blocks() as demo:
     output_plot = gr.Plot(label="MIDI score plot")
     output_midi = gr.File(label="MIDI file", file_types=[".mid"])
 
-    generate_btn.click(Generate_Drums, 
+    generate_btn.click(Humanize_MIDI, 
                        [input_midi,
                         model_temperature,
                         model_sampling_top_p
@@ -462,7 +467,7 @@ with gr.Blocks() as demo:
                  output_plot,
                  output_midi
                 ],
-                Generate_Drums
+                Humanize_MIDI
     )
     
 #==================================================================================