Update app.py

app.py

@@ -222,13 +222,23 @@ def InpaintPitches(input_midi, input_num_of_notes, input_patch_number):
     print('=' * 70)
     print('Giant Music Transformer Inpainting Model Generator')
     print('=' * 70)
+
+    nidx = 0
     
+    for m in enumerate(melody_chords):
+        if 2304 <= melody_chords[i] < 18945 and (cpatch) == inpaint_MIDI_patch:
+            nidx += 1
+
+        if nidx == input_num_of_notes:
+            nidx = i
+            break
+
     out2 = []
     
     for m in melody_chords[:number_of_prime_tokens]:
       out2.append(m)
     
-    for i in range(number_of_prime_tokens, len(melody_chords[:input_num_of_notes])):
+    for i in range(number_of_prime_tokens, len(melody_chords[:nidx])):
     
         cpatch = (melody_chords[i]-2304) // 129