Update app.py

app.py

@@ -112,7 +112,7 @@ def update_selection(evt: gr.SelectData, width, height):
     )
 
 @spaces.GPU(duration=70)
-def generate_image(prompt, negative_prompt, trigger_word, steps, seed, cfg_scale, width, height, lora_scale):
+def generate_image(prompt, negative_prompt, lora_scale, steps, seed, cfg_scale, width, height):
     pipe.to("cuda")
     generator = torch.Generator(device="cuda").manual_seed(seed)
     
@@ -130,7 +130,7 @@ def generate_image(prompt, negative_prompt, trigger_word, steps, seed, cfg_scale
         ).images[0]
     return image
 
-def run_lora(prompt, negative_prompt, cfg_scale, steps, selected_index, seed, width, height, lora_scale):
+def run_lora(prompt, negative_prompt, lora_scale, cfg_scale, steps, selected_index, seed, width, height):
     if negative_prompt == "":
         negative_prompt = None
     if selected_index is None:
@@ -167,7 +167,7 @@ def run_lora(prompt, negative_prompt, cfg_scale, steps, selected_index, seed, wi
     #        pipe.load_lora_weights(lora_path, adapter_name=selected_lora["repo"], lora_scale=[1.0])
      #       pipe.set_adapters(["fast", selected_lora["repo"]], adapter_weights=[1.0, 1.0])
 
-    image = generate_image(prompt, negative_prompt, steps, seed, cfg_scale, width, height, lora_scale)
+    image = generate_image(prompt, lora_scale, negative_prompt, steps, seed, cfg_scale, width, height)
     pipe.to("cpu")
     pipe.unload_lora_weights()
     return image, seed  
@@ -246,7 +246,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
     gr.on(
         triggers=[generate_button.click, prompt.submit],
         fn=run_lora,
-        inputs=[prompt, negative_prompt, cfg_scale, steps, selected_index, seed, width, height, lora_scale],
+        inputs=[prompt, negative_prompt, lora_scale, cfg_scale, steps, selected_index, seed, width, height],
         outputs=[result, seed]
     )