Update app.py

app.py

@@ -187,6 +187,9 @@ def load_midi(input_midi):
 
 @spaces.GPU
 def Humanize_MIDI(input_midi,
+                  num_prime_toks,
+                  humanize_durations,
+                  humanize_velocities,
                   model_temperature,
                   model_sampling_top_p
                  ):
@@ -204,6 +207,9 @@ def Humanize_MIDI(input_midi,
     fn = os.path.basename(input_midi)
     fn1 = fn.split('.')[0]
     print('Input MIDI file name:', fn)
+    print('Number of prime tokens:', num_prime_toks)
+    print('Humanize durations:', humanize_durations)
+    print('Humanize velocities:', humanize_velocities)
     print('Model temperature:', model_temperature)
     print('Model top p:', model_sampling_top_p)
    
@@ -224,21 +230,18 @@ def Humanize_MIDI(input_midi,
         
             #==================================================================
 
-            dur_vel_toks_num = len([t for t in score if 16767 < t < 18816])
+            dur_vel_toks_num = len([t for t in score[num_prime_toks:] if 16767 < t < 18816])
 
-            print('Number of tokens to inapint:', dur_vel_toks_num)
+            print('Number of tokens to humanize:', dur_vel_toks_num)
 
             #==================================================================
             
             print('=' * 70)
             print('Generating...')
 
-            inpaint_durations = True
-            inpaint_velocities = True
-
-            final_song = []
+            final_song = score[:num_prime_toks]
         
-            for t in tqdm.tqdm(score):
+            for t in tqdm.tqdm(score[:num_prime_toks]):
 
                 if t < 16767 or t > 18815:
                     final_song.append(t)
@@ -265,10 +268,10 @@ def Humanize_MIDI(input_midi,
                     gdur = ((y-16768) // 8)
                     gvel = (((y-16768) % 8)+1)                    
                     
-                    if inpaint_durations:
+                    if humanize_durations:
                         fdur = gdur
 
-                    if inpaint_velocities:
+                    if humanize_velocities:
                         fvel = gvel
                         
                     dur_vel_tok = ((8 * fdur) + fvel) + 16768
@@ -447,6 +450,9 @@ with gr.Blocks() as demo:
 
     generate_btn.click(Humanize_MIDI, 
                        [input_midi,
+                        num_prime_toks,
+                        humanize_durations,
+                        humanize_velocities,
                         model_temperature,
                         model_sampling_top_p
                        ], 
@@ -460,6 +466,9 @@ with gr.Blocks() as demo:
                 [["Sharing The Night Together.kar", 0.9, 0.96]
                 ],
                 [input_midi,
+                 num_prime_toks,
+                 humanize_durations,
+                 humanize_velocities,
                  model_temperature,
                  model_sampling_top_p
                 ],