Update app.py

app.py

@@ -253,6 +253,12 @@ def save_midi(tokens, batch_number=None):
 
     return song_f
 
+#==================================================================================
+
+    final_composition = []
+    generated_batches = []
+    block_lines = []
+
 #==================================================================================
 
 @spaces.GPU
@@ -469,7 +475,7 @@ def add_batch(batch_number):
 
 def remove_batch(batch_number, num_tokens):
 
-    # global final_composition
+    global final_composition
 
     if len(final_composition) > num_tokens:
         final_composition = final_composition[:-num_tokens]
@@ -500,9 +506,9 @@ def remove_batch(batch_number, num_tokens):
 
 def reset():
     
-    # global final_composition
-    # global generated_batches
-    # global block_lines
+    global final_composition
+    global generated_batches
+    global block_lines
     
     final_composition = []
     generated_batches = []
@@ -539,12 +545,6 @@ with gr.Blocks() as demo:
 
     #==================================================================================
 
-    final_composition = gr.State([])
-    generated_batches = gr.State([])
-    block_lines = gr.State([])
-
-    #==================================================================================
-
     gr.Markdown("## Upload seed MIDI or click 'Generate' button for random output")
     
     input_midi = gr.File(label="Input MIDI", file_types=[".midi", ".mid", ".kar"])