Update app.py

app.py

@@ -473,6 +473,8 @@ print('=' * 70)
 
 with gr.Blocks() as demo:
 
+    demo.load(reset)
+
     gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Giant Music Transformer</h1>")
     gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Fast multi-instrumental music transformer with true full MIDI instruments range, efficient encoding, octo-velocity and outro tokens</h1>")
     gr.Markdown(
@@ -543,7 +545,7 @@ with gr.Blocks() as demo:
                      outputs=[final_audio_output, final_plot_output, final_file_output]                     
                     )
 
-    demo.unload(lambda: print("User ended session."))
+    demo.unload(reset)
 
 #==================================================================================