Update app.py

app.py

@@ -213,8 +213,7 @@ def InpaintPitches(input_midi, input_num_of_notes, input_patch_number):
     inpaint_MIDI_patch = input_patch_number
     
     #@markdown Generation settings
-    
-    number_of_prime_tokens = 90 # @param {type:"slider", min:3, max:8190, step:3}
+    number_of_prime_notes = 7
     number_of_memory_tokens = 1024 # @param {type:"slider", min:3, max:8190, step:3}
     number_of_samples_per_inpainted_note = 1 #@param {type:"slider", min:1, max:16, step:1}
     temperature = 0.85
@@ -223,7 +222,11 @@ def InpaintPitches(input_midi, input_num_of_notes, input_patch_number):
     print('Giant Music Transformer Inpainting Model Generator')
     print('=' * 70)
 
+    #==========================================================================
+
     nidx = 0
+    first_inote = True
+    fidx = 0
     
     for i, m in enumerate(melody_chords):
         
@@ -231,11 +234,19 @@ def InpaintPitches(input_midi, input_num_of_notes, input_patch_number):
         
         if 2304 <= melody_chords[i] < 18945 and (cpatch) == inpaint_MIDI_patch:
             nidx += 1
+            if first_inote:
+                fidx += 1
 
-        if nidx == input_num_of_notes+(number_of_prime_tokens // 3):
+            if fidx == number_of_prime_notes:
+                number_of_prime_tokens = i
+                first_inote = False
+
+        if nidx == input_num_of_notes:
             break
             
     nidx = i
+
+    #==========================================================================
     
     out2 = []