Update app.py

app.py

@@ -533,7 +533,7 @@ with gr.Blocks() as demo:
             
             outputs.extend([audio_output, plot_output])
 
-    outputs.extend([[]])
+    outputs.extend([model_state])
 
     generate_btn.click(generate_callback_wrapper, 
                        [input_midi,