Update app.py

app.py

@@ -156,46 +156,30 @@ with gr.Blocks(css=css) as demo:
         )
         
         # PART 1 • MODELS
-        with gr.Row():
+        if not is_shared_ui:
+            your_username = api.whoami()["name"]
+            my_models = api.list_models(author=your_username, filter=["diffusers", "stable-diffusion-xl", 'lora'])
+            model_names = [item.modelId for item in my_models]
             
-            with gr.Column():
-
-                if not is_shared_ui:
-
-                    your_username = api.whoami()["name"]
-                    my_models = api.list_models(author=your_username, filter=["diffusers", "stable-diffusion-xl", 'lora'])
-                    model_names = [item.modelId for item in my_models]
-                    
-                    #print(model_names)           
-                    
+            #print(model_names)           
+            
+            with gr.Row():
                 
+                with gr.Column():
+            
                     lora_1_id = gr.Dropdown(
                         label = "LoRa 1 ID",
                         choices = model_names,
                         allow_custom_value = True
                         #placeholder = "username/model_id"
                     )
-                
-                else:
-                    lora_1_id = gr.Textbox(
-                        label = "LoRa 1 ID",
-                        placeholder = "username/model_id"
-                    )
-
-                lora_1_sfts = gr.Dropdown(
-                    label = "Safetensors file",
-                    visible=False
-                )
-            
-            with gr.Column():
-
-                if not is_shared_ui:
-
-                    your_username = api.whoami()["name"]
-                    my_models = api.list_models(author=your_username, filter=["diffusers", "stable-diffusion-xl", 'lora'])
-                    model_names = [item.modelId for item in my_models]
                     
-                    #print(model_names)           
+                    lora_1_sfts = gr.Dropdown(
+                        label = "Safetensors file",
+                        visible=False
+                    )
+                
+                with gr.Column():
                     
                     lora_2_id = gr.Dropdown(
                         label = "LoRa 2 ID",
@@ -203,18 +187,36 @@ with gr.Blocks(css=css) as demo:
                         allow_custom_value = True
                         #placeholder = "username/model_id"
                     )
+    
+                    lora_2_sfts = gr.Dropdown(
+                        label = "Safetensors file",
+                        visible=False
+                    )
+        else:
+            with gr.Row():
                 
-                else:
-                    lora_2_id = gr.Textbox(
-                        label = "LoRa 2 ID",
+                with gr.Column():
+                    lora_1_id = gr.Textbox(
+                        label = "LoRa 1 ID",
                         placeholder = "username/model_id"
                     )
 
-                lora_2_sfts = gr.Dropdown(
-                    label = "Safetensors file",
-                    visible=False
-                )
+                    lora_1_sfts = gr.Dropdown(
+                        label = "Safetensors file",
+                        visible=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
+                    )
+        
         load_models_btn = gr.Button("Load models and .safetensors")
 
         # PART 2 • INFERENCE