Update app.py

app.py

@@ -214,10 +214,10 @@ def InpaintPitches(input_midi, input_num_of_notes, input_patch_number):
     
     #@markdown Generation settings
     
-    number_of_prime_tokens = 30 # @param {type:"slider", min:3, max:8190, step:3}
+    number_of_prime_tokens = 90 # @param {type:"slider", min:3, max:8190, step:3}
     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.9
+    temperature = 0.85
     
     print('=' * 70)
     print('Giant Music Transformer Inpainting Model Generator')
@@ -232,10 +232,11 @@ 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 nidx == input_num_of_notes:
-            nidx = i
+        if nidx == input_num_of_notes+(number_of_prime_tokens // 3):
             break
-
+            
+    nidx = i
+    
     out2 = []
     
     for m in melody_chords[:number_of_prime_tokens]: