Update app.py

app.py

@@ -226,9 +226,7 @@ def GenerateAccompaniment(input_midi, input_num_tokens, input_conditioning_type,
         if input_conditioning_type == 'Chords-Times-Durations':
           output.append(durs[idx])
 
-        chords = output
-    
-        x = torch.tensor([chords] * 1, dtype=torch.long, device='cuda')
+        x = torch.tensor([output] * 1, dtype=torch.long, device='cuda')
         
         o = 0
         
@@ -256,7 +254,7 @@ def GenerateAccompaniment(input_midi, input_num_tokens, input_conditioning_type,
     
         idx += 1
 
-        if idx == len(chords[:input_num_tokens])-1:
+        if idx == len(chords[:input_num_tokens]):
             break
 
     print('=' * 70)