Update app.py

app.py

@@ -217,83 +217,59 @@ with gr.Blocks(css=css) as demo:
             
             #print(model_names)           
             
-            with gr.Row():
+        with gr.Row():
                 
-                with gr.Column():
-            
+            with gr.Column():
+
+                if not is_shared_ui:
                     lora_1_id = gr.Dropdown(
                         label = "LoRa 1 ID",
                         choices = model_names,
                         allow_custom_value = True
                         #placeholder = "username/model_id"
                     )
-                    
-                    lora_1_sfts = gr.Dropdown(
-                        label = "Safetensors file",
-                        visible=False
+                else:
+                    lora_1_id = gr.Textbox(
+                        label = "LoRa 1 ID",
+                        placeholder = "username/model_id"
                     )
+                    
+                lora_1_sfts = gr.Dropdown(
+                    label = "Safetensors file",
+                    visible=False
+                )
 
-                    instance_prompt_1 = gr.Textbox(
-                        label = "Trigger Word 1",
-                        visible = False,
-                        interactive = False
-                    )
+                instance_prompt_1 = gr.Textbox(
+                    label = "Trigger Word 1",
+                    visible = False,
+                    interactive = False
+                )
                 
-                with gr.Column():
-                    
+            with gr.Column():
+
+                if not is_shared_ui:
                     lora_2_id = gr.Dropdown(
                         label = "LoRa 2 ID",
                         choices = model_names,
                         allow_custom_value = True
                         #placeholder = "username/model_id"
                     )
-    
-                    lora_2_sfts = gr.Dropdown(
-                        label = "Safetensors file",
-                        visible=False
-                    )
-
-                    instance_prompt_2 = gr.Textbox(
-                        label = "Trigger Word 2",
-                        visible = False,
-                        interactive = False
-                    )
-        else:
-            with gr.Row():
-                
-                with gr.Column():
-                    lora_1_id = gr.Textbox(
-                        label = "LoRa 1 ID",
+                else:
+                    lora_2_id = gr.Textbox(
+                        label = "LoRa 2 ID",
                         placeholder = "username/model_id"
                     )
-
-                    lora_1_sfts = gr.Dropdown(
-                        label = "Safetensors file",
-                        visible=False
-                    )
-
-                    instance_prompt_1 = gr.Textbox(
-                        label = "Trigger Word 1",
-                        visible = False,
-                        interactive = False
-                    )
-
-                with gr.Column():
-                    lora_2_id = gr.Textbox(
-                            label = "LoRa 2 ID",
-                            placeholder = "username/model_id"
-                        )
     
-                    lora_2_sfts = gr.Dropdown(
-                        label = "Safetensors file",
-                        visible=False
-                    )
+                lora_2_sfts = gr.Dropdown(
+                    label = "Safetensors file",
+                    visible=False
+                )
 
-                    instance_prompt_2 = gr.Textbox(
-                        label = "Trigger Word 2",
-                        visible = False,
-                        interactive = False
-                    )
+                instance_prompt_2 = gr.Textbox(
+                    label = "Trigger Word 2",
+                    visible = False,
+                    interactive = False
+                )
         
         load_models_btn = gr.Button("1. Load models and .safetensors")