Update app.py

app.py

@@ -205,38 +205,36 @@ def Generate_Accompaniment(input_midi,
 
     def generate_block_seq(input_seq, trg_dtime, temperature=0.9):
     
+        inp_seq = copy.deepcopy(input_seq)
+    
+        block_seq = []
+    
         cur_time = 0
     
-        block_seq = [128]
+        while cur_time < trg_dtime:
+    
+            x = torch.LongTensor(inp_seq).cuda()
+            
+            with ctx:
+                out = model.generate(x,
+                                     1,
+                                     temperature=temperature,
+                                     return_prime=False,
+                                     verbose=False)
+            
+            y = out.tolist()[0][0]
+    
+            if y < 128:
+                cur_time += y
     
-        while cur_time != trg_dtime or len(block_seq) < 2 or block_seq[-1] > 127:
+            inp_seq.append(y)
+            block_seq.append(y)
     
-            inp_seq = copy.deepcopy(input_seq)
-        
-            block_seq = []
-        
-            cur_time = 0
-        
-            while cur_time < trg_dtime:
-        
-                x = torch.LongTensor(inp_seq).cuda()
-                
-                with ctx:
-                    out = model.generate(x,
-                                         1,
-                                         temperature=temperature,
-                                         return_prime=False,
-                                         verbose=False)
-                
-                y = out.tolist()[0][0]
-        
-                if y < 128:
-                    cur_time += y
-        
-                inp_seq.append(y)
-                block_seq.append(y)
+        if cur_time != trg_dtime:
+            return []
     
-        return block_seq
+        else:
+            return block_seq
 
     #===============================================================================