Update app.py

app.py

@@ -297,7 +297,7 @@ def Generate_Chords(input_midi, input_num_prime_chords, input_num_gen_chords, in
     for m in melody_chords2[:input_num_prime_chords]:
         output.extend(m)
     
-    for ct in tqdm.tqdm(melody_chords2[input_num_prime_chords:]):
+    for ct in tqdm.tqdm(melody_chords2[input_num_prime_chords:input_num_gen_chords]):
 
         if input_conditioning_type == 'Chords':
             output.extend(ct[:1])
@@ -309,7 +309,7 @@ def Generate_Chords(input_midi, input_num_prime_chords, input_num_gen_chords, in
     
         while y > 645:
     
-            x = torch.LongTensor(song).to(DEVICE)
+            x = torch.LongTensor(output).to(DEVICE)
             
             with ctx:
                 out = model.generate(x,