Update app.py

app.py

@@ -213,7 +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_notes = 7
+    number_of_prime_notes = 12
     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
@@ -227,19 +227,23 @@ def InpaintPitches(input_midi, input_num_of_notes, input_patch_number):
     nidx = 0
     first_inote = True
     fidx = 0
+
+    number_of_prime_tokens = number_of_prime_notes * 3
     
     for i, m in enumerate(melody_chords):
-        
-        cpatch = (melody_chords[i]-2304) // 129
-        
-        if 2304 <= melody_chords[i] < 18945 and (cpatch) == inpaint_MIDI_patch:
-            nidx += 1
-            if first_inote:
-                fidx += 1
-
-            if fidx == number_of_prime_notes:
-                number_of_prime_tokens = i
-                first_inote = False
+    
+        if 2304 <= melody_chords[i] < 18945:
+            
+            cpatch = (melody_chords[i]-2304) // 129
+            
+            if cpatch == inpaint_MIDI_patch:
+                nidx += 1
+                if first_inote:
+                    fidx += 1
+    
+                if first_inote and fidx == number_of_prime_notes:
+                    number_of_prime_tokens = i
+                    first_inote = False
 
         if nidx == input_num_of_notes:
             break