Update app.py

app.py

@@ -233,6 +233,11 @@ def Generate_Rock_Song(input_midi,
     
         gen_tokens = []
 
+        seen = False
+    
+        if 256 < input[-1] < 544:
+          seen = True 
+
         while pcount < max_num_ptcs and y > 255 and tries < max_tries:
     
             x = torch.tensor(input[-input_num_memory_tokens:], dtype=torch.long, device=DEVICE)
@@ -248,14 +253,23 @@ def Generate_Rock_Song(input_midi,
             
             y = out[0].tolist()[0]
     
-            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
+            if 256 < y < 544:
+                if not seen:
+                    input.append(y)
+                    gen_tokens.append(y)
+                    seen = True
                     
-            else:
-                tries += 1
+                else:
+                    tries += 1
+                    
+            if y > 544 and seen:
+                if pcount < max_num_ptcs and y not in gen_tokens:    
+                    input.append(y)
+                    gen_tokens.append(y)
+                    pcount += 1
+    
+                else:
+                    tries += 1
                 
         return gen_tokens