Update app.py

app.py

@@ -318,12 +318,6 @@ def generate_music(prime,
     
 #==================================================================================
 
-final_composition = []
-generated_batches = []
-block_lines = []
-
-#==================================================================================
-
 def generate_callback(input_midi, 
                       num_prime_tokens, 
                       num_gen_tokens,
@@ -334,7 +328,7 @@ def generate_callback(input_midi,
                       model_sampling_top_p
                      ):
 
-    global generated_batches
+    # global generated_batches
     generated_batches = []
 
     if not final_composition and input_midi is not None:
@@ -475,7 +469,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]
@@ -506,9 +500,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 = []
@@ -543,6 +537,14 @@ with gr.Blocks() as demo:
             for faster execution and endless generation!
             """)
 
+    #==================================================================================
+
+    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"])