Update app.py

app.py

@@ -173,7 +173,7 @@ with gr.Blocks(css=css) as demo:
 <h2 style="text-align: center;">SD-XL Custom Model Inference</h2>
 <p style="text-align: center;">Use this demo to check results from your previously trained LoRa model.</p>
         """)
-        with gr.Box():
+        with gr.Group():
             with gr.Row():
                 with gr.Column():
                     if not is_shared_ui: