Update app.py

app.py

@@ -223,20 +223,19 @@ def Generate_Rock_Song(input_midi,
 
     #==================================================================
 
-    def generate_tokens(seq, max_num_ptcs=10):
+    def generate_tokens(seq, max_num_ptcs=4, max_tries=10):
     
         input = copy.deepcopy(seq)
 
-        input = input[-input_num_memory_tokens:]
-    
         pcount = 0
         y = 545
+        tries = 0
     
         gen_tokens = []
+
+        while pcount < max_num_ptcs and y > 255 and tries < max_tries:
     
-        while pcount < max_num_ptcs and y > 255:
-    
-            x = torch.tensor(input, dtype=torch.long, device=DEVICE)
+            x = torch.tensor(input[-input_num_memory_tokens:], dtype=torch.long, device=DEVICE)
         
             with ctx:
               out = model.generate(x,
@@ -249,12 +248,15 @@ def Generate_Rock_Song(input_midi,
             
             y = out[0].tolist()[0]
     
-            if pcount < max_num_ptcs and y > 255:
+            if pcount < max_num_ptcs and y > 255 and y not in gen_tokens:
                 input.append(y)
                 gen_tokens.append(y)
                 if y > 544:
                     pcount += 1
-    
+                    
+            else:
+                tries += 1
+                
         return gen_tokens
 
     #==================================================================