replace gr.Box with gr.Group

app.py

@@ -253,7 +253,7 @@ with gr.Blocks(css=css) as demo:
 
         use_custom_model = gr.Checkbox(label="Use a custom pre-trained LoRa model ? (optional)", value=False, info="To use a private model, you'll need to duplicate the space with your own access token.")
         
-        with gr.Box(visible=False) as custom_model_box:
+        with gr.Group(visible=False) as custom_model_box:
             with gr.Row():
                 with gr.Column():
                     if not is_shared_ui: