Update main.py

main.py

@@ -79,7 +79,7 @@ def run():
                     btn_remove_last = gr.Button("↩️ Remove last instrument")
                     btn_regenerate_last = gr.Button("🔄 Regenerate last instrument")
             with gr.Column():
-                with gr.Box():
+                with gr.Group():
                     audio_output = gr.Video(show_share_button=True)
                     midi_file = gr.File()
                     with gr.Row():