Update app.py

app.py

@@ -163,7 +163,7 @@ def create_demo(
         
         with gr.Row():
             with gr.Column(scale=3):
-                with gr.Box(elem_classes="container"):
+                with gr.Group(elem_classes="container"):
                     prompt = gr.Textbox(
                         label="Prompt",
                         placeholder="Describe the image you want to generate...",
@@ -180,7 +180,7 @@ def create_demo(
 
                     with gr.Row():
                         with gr.Column(scale=2):
-                            with gr.Box(elem_classes="slider-container"):
+                            with gr.Group(elem_classes="slider-container"):
                                 width = gr.Slider(512, 2048, 512, step=16, label="Generation Width")
                                 height = gr.Slider(512, 2048, 512, step=16, label="Generation Height")
                         
@@ -199,7 +199,7 @@ def create_demo(
                     generate_btn = gr.Button("Generate", elem_classes="generate-btn")
 
             with gr.Column(scale=2):
-                with gr.Box(elem_classes="output-container"):
+                with gr.Group(elem_classes="output-container"):
                     gr.Markdown("### Generated Result")
                     output_image = gr.Image(label="Generated Image")
                     download_btn = gr.File(label="Download full-resolution", type="filepath", interactive=False)
@@ -218,7 +218,7 @@ def create_demo(
         example_text = gr.Text("", visible=False, label="Case For:")
         examples = get_examples("./assets/examples")
 
-        with gr.Box(elem_classes="container"):
+        with gr.Group(elem_classes="container"):
             gr.Markdown("### <div style='text-align: center; margin-bottom: 1rem;'>Examples</div>")
             gr.Examples(
                 examples=examples,