Update app.py

app.py

@@ -117,9 +117,9 @@ def update_selection(evt: gr.SelectData, selected_indices, loras_state, width, h
             gr.Warning("You can select up to 2 LoRAs, remove one to select a new one.")
             return gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), width, height, gr.update(), gr.update()
 
-    selected_info_1 = "Select a LoRA 1"
+    selected_info_1 = "Select a Celebrity as LoRA 1"
     selected_info_2 = "Select a LoRA 2"
-    lora_scale_1 = 1.15
+    lora_scale_1 = 0.6
     lora_scale_2 = 1.15
     lora_image_1 = None
     lora_image_2 = None
@@ -143,9 +143,9 @@ def update_selection(evt: gr.SelectData, selected_indices, loras_state, width, h
 def remove_lora_1(selected_indices, loras_state):
     if len(selected_indices) >= 1:
         selected_indices.pop(0)
-    selected_info_1 = "Select a LoRA 1"
+    selected_info_1 = "Select a Celebrity as LoRA 1"
     selected_info_2 = "Select a LoRA 2"
-    lora_scale_1 = 1.15
+    lora_scale_1 = 0.6
     lora_scale_2 = 1.15
     lora_image_1 = None
     lora_image_2 = None
@@ -162,9 +162,9 @@ def remove_lora_1(selected_indices, loras_state):
 def remove_lora_2(selected_indices, loras_state):
     if len(selected_indices) >= 2:
         selected_indices.pop(1)
-    selected_info_1 = "Select LoRA 1"
+    selected_info_1 = "Select a Celebrity as LoRA 1"
     selected_info_2 = "Select LoRA 2"
-    lora_scale_1 = 1.15
+    lora_scale_1 = 0.6
     lora_scale_2 = 1.15
     lora_image_1 = None
     lora_image_2 = None
@@ -222,9 +222,9 @@ def add_custom_lora(custom_lora, selected_indices, current_loras, gallery):
                 gr.Warning("You can select up to 2 LoRAs, remove one to select a new one.")
 
             # Update selected_info and images
-            selected_info_1 = "Select a LoRA 1"
+            selected_info_1 = "Select a Celebrity as LoRA 1"
             selected_info_2 = "Select a LoRA 2"
-            lora_scale_1 = 1.15
+            lora_scale_1 = 0.6
             lora_scale_2 = 1.15
             lora_image_1 = None
             lora_image_2 = None
@@ -267,9 +267,9 @@ def remove_custom_lora(selected_indices, current_loras, gallery):
     # Update gallery
     gallery_items = [(item["image"], item["title"]) for item in current_loras]
     # Update selected_info and images
-    selected_info_1 = "Select a LoRA 1"
+    selected_info_1 = "Select a Celebrity as LoRA 1"
     selected_info_2 = "Select a LoRA 2"
-    lora_scale_1 = 1.15
+    lora_scale_1 = 0.6
     lora_scale_2 = 1.15
     lora_image_1 = None
     lora_image_2 = None
@@ -396,7 +396,7 @@ def run_lora(prompt, cfg_scale, steps, selected_indices, lora_scale_1, lora_scal
         image_generator = generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
 # Consume the generator to get the final image
 #        final_image = None
-#        step_counter = 0
+        step_counter = 0
         for image in image_generator:
             step_counter += 1
             final_image = image
@@ -552,7 +552,7 @@ with gr.Blocks(css=css, delete_cache=(240, 240)) as app:
 #                remove_button_4 = gr.Button("Remove", size="sm")
 
     with gr.Row():
-        with gr.Accordion("Advanced Settings", open=False):
+        with gr.Accordion("Advanced Settings", open=True):
             #with gr.Row():
             #    input_image = gr.Image(label="Input image", type="filepath", show_share_button=False)
             #    image_strength = gr.Slider(label="Denoise Strength", info="Lower means more image influence", minimum=0.1, maximum=1.0, step=0.01, value=0.75)
@@ -562,7 +562,7 @@ with gr.Blocks(css=css, delete_cache=(240, 240)) as app:
                     steps = gr.Slider(label="Steps", minimum=1, maximum=50, step=1, value=28)
 
                 with gr.Row():
-                    width = gr.Slider(label="Width", minimum=256, maximum=1536, step=64, value=1024)
+                    width = gr.Slider(label="Width", minimum=256, maximum=1536, step=64, value=768)
                     height = gr.Slider(label="Height", minimum=256, maximum=1536, step=64, value=1024)
 
                 with gr.Row():