Update app.py

app.py

@@ -332,7 +332,7 @@ def generate_music_and_state(input_midi, num_prime_tokens, num_gen_tokens,
         drum_pitch = random.choice([36, 38])
         final_composition.extend([(128*128)+drum_pitch+256]) # Drum token
 
-    batched_gen_tokens = generate_music(final_composition, num_gen_tokens, num_mem_tokens,
+    batched_gen_tokens = generate_music(final_composition, num_gen_tokens,
                                         NUM_OUT_BATCHES, model_temperature, model_top_p)
     
     output_batches = []
@@ -479,7 +479,7 @@ with gr.Blocks() as demo:
             outputs.extend([audio_output, plot_output])
     generate_btn.click(
         generate_music_and_state,
-        [input_midi, num_prime_tokens, num_gen_tokens, num_mem_tokens, model_temperature, model_top_p, add_drums, add_outro,
+        [input_midi, num_prime_tokens, num_gen_tokens, model_temperature, model_top_p, add_drums, add_outro,
          final_composition, generated_batches, block_lines],
         outputs
     )