Update app.py

app.py

@@ -256,6 +256,9 @@ def GenerateAccompaniment(input_midi, input_num_tokens, input_conditioning_type,
     
         idx += 1
 
+        if idx == len(chords[:input_num_tokens]):
+            break
+
     print('=' * 70)
     print('Done!')
     print('=' * 70)