Update main.py

main.py

@@ -71,7 +71,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():  # Replace gr.Box() with gr.Group()
                     audio_output = gr.Video(show_share_button=True)
                     midi_file = gr.File()
                     with gr.Row():
@@ -154,4 +154,4 @@ def run():
 
 
 if __name__ == "__main__":
-    run()
+    run()