Update app.py

app.py

@@ -167,17 +167,67 @@ def Generate_Rock_Song(input_midi, input_melody_seed_number):
     print('=' * 70)
     print('Generating...')
 
-    x = (torch.tensor(seed_melody, dtype=torch.long, device='cuda')[None, ...])
+    #==================================================================
+
+    def generate_tokens(seq, max_num_ptcs=10):
+    
+        input = copy.deepcopy(seq)
+    
+        pcount = 0
+        y = 545
+    
+        gen_tokens = []
+    
+        while pcount < max_num_ptcs and y > 255:
+    
+            x = torch.tensor(input, dtype=torch.long, device='cuda')
+        
+            with ctx:
+              out = model.generate(x,
+                                  1,
+                                  filter_logits_fn=top_k,
+                                  filter_kwargs={'k': 10},
+                                  temperature=0.9,
+                                  return_prime=False,
+                                  verbose=False)
+            
+            y = out[0].tolist()[0]
+    
+            if pcount < max_num_ptcs and y > 255:
+                input.append(y)
+                gen_tokens.append(y)
+                if y > 544:
+                    pcount += 1
+    
+        return gen_tokens
 
-    with ctx:
-      out = model.generate(x,
-                          1536,
-                          temperature=0.9,
-                          return_prime=False,
-                          verbose=False)
+    #==================================================================
+
+    num_prime_chords = 128
+    pass_chan_dur_tok = False
+    match_ptcs_counts = False
+    
+    song = []
     
-    output = out[0].tolist()
+    for i in range(num_prime_chords):
+        song.extend(prime_toks[i])
+    
+    for i in tqdm.tqdm(range(num_prime_chords, len(score_toks))):
         
+        song.extend(score_toks[i])
+    
+        if control_toks[i]:
+            for ct in control_toks[i]:
+                if pass_chan_dur_tok:
+                    song.append(ct[0])
+                if match_ptcs_counts:
+                    out_seq = generate_tokens(song, ct[1])
+                else:
+                    out_seq = generate_tokens(song)
+                song.extend(out_seq)
+
+    #==================================================================
+            
     print('=' * 70)
     print('Done!')
     print('=' * 70)