Update app.py

app.py

@@ -143,7 +143,7 @@ def check_selected(selected_state):
     if not selected_state:
         raise gr.Error("You must select a LoRA")
 
-def run_lora(prompt, negative, lora_scale, selected_state, sdxl_loras, sdxl_loras_new, progress=gr.Progress(track_tqdm=True)):
+def run_lora(prompt, negative, lora_scale, steps, selected_state, sdxl_loras, sdxl_loras_new, progress=gr.Progress(track_tqdm=True)):
     global last_lora, last_merged, last_fused, pipe
     print("Index when running ", selected_state.index)
     if(selected_state.index < 0):
@@ -185,7 +185,8 @@ def run_lora(prompt, negative, lora_scale, selected_state, sdxl_loras, sdxl_lora
         prompt=prompt,
         negative_prompt=negative,
         num_inference_steps=4,
-        guidance_scale=0.5
+        guidance_scale=0.5,
+        num_inference_steps=steps
         
     ).images[0]
     
@@ -211,7 +212,7 @@ with gr.Blocks(css="custom.css") as demo:
     gr_sdxl_loras = gr.State(value=sdxl_loras_raw)
     gr_sdxl_loras_new = gr.State(value=sdxl_loras_raw_new)
     title = gr.HTML(
-        """<h1><img src="https://i.imgur.com/vT48NAO.png" alt="LoRA"><small><small><i>Experimental</i></small></small> LCM LoRA the Explorer</h1>
+        """<h1><img src="https://i.imgur.com/vT48NAO.png" alt="LoRA"><small><small><i>Experimental</i></small></small>&nbsp;&nbsp;LCM LoRA the Explorer</h1>
         <br>
         Combine loading an <a href="#">LCM-LoRA</a> with your favorite SDXL LoRa and run LoRAs in only 4 steps. Check out <a href="#">our blog</a>on how this works. <b>Not all LoRAs may combine well with the LCM-LoRA</b>
         """,
@@ -249,8 +250,9 @@ with gr.Blocks(css="custom.css") as demo:
                 interactive=False, label="Generated Image", elem_id="result-image"
             )
             with gr.Accordion("Advanced options", open=False):
-                negative = gr.Textbox(label="Negative Prompt")
+                negative = gr.Textbox(label="Negative Prompt", visible=False)
                 weight = gr.Slider(0, 10, value=0.8, step=0.1, label="LoRA weight")
+                steps = gr.Slider(0,8, value=4, step=1, label="Steps")
     with gr.Column(elem_id="extra_info"):
         with gr.Accordion(
             "Use it with: 🧨  diffusers, ComfyUI, Invoke AI, SD.Next, AUTO1111",
@@ -315,7 +317,7 @@ with gr.Blocks(css="custom.css") as demo:
         show_progress=False
     ).success(
         fn=run_lora,
-        inputs=[prompt, negative, weight, selected_state, gr_sdxl_loras, gr_sdxl_loras_new],
+        inputs=[prompt, negative, weight, steps, selected_state, gr_sdxl_loras, gr_sdxl_loras_new],
         outputs=[result, share_group],
     )
     button.click(
@@ -325,7 +327,7 @@ with gr.Blocks(css="custom.css") as demo:
         show_progress=False
     ).success(
         fn=run_lora,
-        inputs=[prompt, negative, weight, selected_state, gr_sdxl_loras, gr_sdxl_loras_new],
+        inputs=[prompt, negative, weight, steps, selected_state, gr_sdxl_loras, gr_sdxl_loras_new],
         outputs=[result, share_group],
     )
     share_button.click(None, [], [], js=share_js)