Update app.py

app.py

@@ -202,12 +202,6 @@ def InpaintPitches(input_midi, input_num_of_notes, input_patch_number):
     print('=' * 70)
     print('Generating...')
 
-    output = []
-    
-    max_chords_limit = 8
-    temperature=0.9
-    num_memory_tokens=4096
-
     #@title Pitches/Instruments Inpainting
     
     #@markdown You can stop the inpainting at any time to render partial results
@@ -220,8 +214,8 @@ def InpaintPitches(input_midi, input_num_of_notes, input_patch_number):
     
     #@markdown Generation settings
     
-    number_of_prime_tokens = 90 # @param {type:"slider", min:3, max:8190, step:3}
-    number_of_memory_tokens = 2048 # @param {type:"slider", min:3, max:8190, step:3}
+    number_of_prime_tokens = 30 # @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}
     
     print('=' * 70)
@@ -233,7 +227,7 @@ def InpaintPitches(input_midi, input_num_of_notes, input_patch_number):
     for m in melody_chords[:number_of_prime_tokens]:
       out2.append(m)
     
-    for i in range(number_of_prime_tokens, len(melody_chords)):
+    for i in range(number_of_prime_tokens, len(melody_chords[:input_num_of_notes])):
     
         cpatch = (melody_chords[i]-2304) // 129