Update app.py

app.py

@@ -18,7 +18,7 @@ in_space = os.getenv("SYSTEM") == "spaces"
 #=================================================================================================
 
 @torch.no_grad()
-def GenerateMIDI():
+def GenerateMIDI(progress=gr.Progress()):
 
     start_tokens = [3087, 3073+1, 3075+1]
     seq_len = 512
@@ -26,7 +26,7 @@ def GenerateMIDI():
     temperature = 1.0
     verbose=False
     return_prime=False
-    progress=gr.Progress()
+    
 
     out = torch.FloatTensor([start_tokens])
 
@@ -58,15 +58,13 @@ def GenerateMIDI():
 
             step += 1
         
-        if step >= seq_len:
-            break
+            if step >= seq_len:
+                break
             
         except Exception as e:
             print('Error', e)
             break
 
-
-            
     if return_prime:
       melody_chords_f = out[:, :]