Update app.py

app.py

@@ -276,7 +276,7 @@ with gr.Blocks(css=css) as demo:
         
         use_custom_model = gr.Checkbox(label="Use a public custom model ?(optional)", value=False, info="To use a private model, you'll prefer to duplicate the space with your own access token.")
         
-        with gr.Box():
+        with gr.Box(visible=True) as custom_model_box:
             with gr.Row():
                 with gr.Column():
                     if not is_shared_ui:
@@ -318,8 +318,9 @@ with gr.Blocks(css=css) as demo:
                             elem_id = "status_info"
                         )
                     trigger_word = gr.Textbox(label="Trigger word", interactive=False, visible=False)
-        
-            custom_lora_weight = gr.Slider(label="Custom model weights", minimum=0.1, maximum=0.9, step=0.1, value=0.9)
+            
+            with gr.Column():
+                custom_lora_weight = gr.Slider(label="Custom model weights", minimum=0.1, maximum=0.9, step=0.1, value=0.9)
         
         submit_btn = gr.Button("Submit")