Update app.py

app.py

@@ -61,7 +61,7 @@ def create_app():
             )
 
         # Dataset Selection
-        with gr.Box():
+        with gr.Group():
             gr.Markdown("## Dataset Selection")
             dataset_source = gr.Radio(
                 label="Dataset Source",
@@ -104,7 +104,7 @@ def create_app():
             )
 
         # Column Mapping
-        with gr.Box():
+        with gr.Group():
             gr.Markdown("## Column Mapping")
             image_column = gr.Textbox(
                 label="Image Column Name",
@@ -118,7 +118,7 @@ def create_app():
             )
 
         # Training Parameters
-        with gr.Box():
+        with gr.Group():
             gr.Markdown("## Training Parameters")
             with gr.Row():
                 num_train_epochs = gr.Slider(