Update app.py

app.py

@@ -22,6 +22,8 @@ in_space = os.getenv("SYSTEM") == "spaces"
 @torch.no_grad()
 def GenerateMIDI(idrums, iinstr, progress=gr.Progress()):
 
+    number_of_notes_to_generate = 140
+
     if idrums:
         drums = 3074
     else:
@@ -36,17 +38,21 @@ def GenerateMIDI(idrums, iinstr, progress=gr.Progress()):
     print(start_tokens)
     print('=' * 70)
     
-    inp = [start_tokens]
-    
-    inp = torch.LongTensor(inp).cpu()
-    
-    out = model.module.generate(inp,
-                          256,
-                          temperature=0.9,
-                          return_prime=False,
-                          verbose=True)
+    outy = start_tokens
+
+    for i in progress.tqdm(range(number_of_notes_to_generate)):
     
-    melody_chords_f = out[0].tolist()
+        inp = torch.LongTensor([outy]).cpu()
+        
+        out = model.module.generate(inp,
+                              3,
+                              temperature=0.9,
+                              return_prime=False,
+                              verbose=True)
+        
+        out0 = out[0].tolist()
+
+        outy.extend(out0)
     
     print('=' * 70)
     print('Done!')