Update app.py

app.py

@@ -186,6 +186,7 @@ def load_midi(input_midi):
 @spaces.GPU
 def Humanize_MIDI(input_midi,
                   num_prime_toks,
+                  num_hum_notes,
                   humanize_durations,
                   humanize_velocities,
                   model_temperature,
@@ -206,6 +207,7 @@ def Humanize_MIDI(input_midi,
     fn1 = fn.split('.')[0]
     print('Input MIDI file name:', fn)
     print('Number of prime tokens:', num_prime_toks)
+    print('Number of notes to humanize:', num_hum_notes)
     print('Humanize durations:', humanize_durations)
     print('Humanize velocities:', humanize_velocities)
     print('Model temperature:', model_temperature)
@@ -238,6 +240,8 @@ def Humanize_MIDI(input_midi,
             print('Generating...')
 
             final_song = score[:num_prime_toks]
+
+            hn_count = 0
         
             for t in tqdm.tqdm(score[num_prime_toks:]):
 
@@ -276,6 +280,11 @@ def Humanize_MIDI(input_midi,
                     
                     final_song.append(dur_vel_tok)
 
+                    hn_count += 1
+
+                    if hn_count == num_hum_notes:
+                        break
+
             #==================================================================
            
             print('=' * 70)
@@ -428,11 +437,12 @@ with gr.Blocks() as demo:
                         )
     
     gr.Markdown("## Generation options")
-
-    num_prime_toks = gr.Slider(0, 1024, value=0, 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")
+    
+    num_prime_toks = gr.Slider(0, 1024, value=0, step=1, label="Number of prime tokens")
+    num_hum_notes = gr.Slider(128, 2048, value=512, step=1, label="Number of notes to humanize")
 
     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")
@@ -449,6 +459,7 @@ with gr.Blocks() as demo:
     generate_btn.click(Humanize_MIDI, 
                        [input_midi,
                         num_prime_toks,
+                        num_hum_notes,
                         humanize_durations,
                         humanize_velocities,
                         model_temperature,
@@ -461,10 +472,11 @@ with gr.Blocks() as demo:
                       )
 
     gr.Examples(
-                [["Sharing The Night Together.kar", 0, False, True, 0.9, 0.96]
+                [["Sharing The Night Together.kar", 0, 1024, False, True, 0.9, 0.96]
                 ],
                 [input_midi,
                  num_prime_toks,
+                 num_hum_notes,
                  humanize_durations,
                  humanize_velocities,
                  model_temperature,