Update app.py

app.py

@@ -49,7 +49,7 @@ def update_selection(evt: gr.SelectData):
     )
 
 @spaces.GPU(duration=80)
-def generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, generator, lora_scale, progress):
+def generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, lora_scale, progress):
     
     generator = torch.Generator(device="cuda").manual_seed(seed)
     
@@ -86,7 +86,7 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
         if randomize_seed:
             seed = random.randint(0, MAX_SEED)
     
-    image = generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, generator, lora_scale, progress)
+    image = generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, lora_scale, progress)
 
     pipe.unload_lora_weights()
     return image, seed