Update app.py

app.py

@@ -206,12 +206,12 @@ def GenerateAccompaniment(input_midi, input_num_tokens, input_acc_type):
     output1 = []
     output2 = []
 
-    cur_time = 0
+    ctime = 0
 
     for m in melody_chords[:input_num_tokens]:
 
         mel = copy.deepcopy(m)
-        mel[0] = mel[0]-cur_time
+        mel[0] = mel[0]-ctime
     
         output1.extend(mel)
 
@@ -225,6 +225,7 @@ def GenerateAccompaniment(input_midi, input_num_tokens, input_acc_type):
         time = input_seq[-2]-512
     
         cur_time = 0
+        ctime = 0
     
         for _ in range(num_toks_per_note):
           with ctx:
@@ -240,6 +241,8 @@ def GenerateAccompaniment(input_midi, input_num_tokens, input_acc_type):
             cur_time += o
     
           if cur_time < time and o < 384:
+              
+            ctime = cur_time
     
             out = torch.LongTensor([[o]]).cuda()
             x = torch.cat((x, out), 1)