Update app.py

app.py

@@ -109,7 +109,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, progress=gr.Progress(track_tqdm=True)):
+def generate_image(prompt, negative_prompt, trigger_word, steps, seed, cfg_scale, width, height, lora_scale):
     pipe.to("cuda")
     generator = torch.Generator(device="cuda").manual_seed(seed)
     
@@ -123,12 +123,11 @@ def generate_image(prompt, negative_prompt, trigger_word, steps, seed, cfg_scale
             width=width,
             height=height,
             generator=generator,
-            progress=gr.Progress(track_tqdm=True),
             joint_attention_kwargs={"scale": lora_scale},
         ).images[0]
     return image
 
-def run_lora(prompt, negative_prompt, cfg_scale, steps, selected_index, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
+def run_lora(prompt, negative_prompt, cfg_scale, steps, selected_index, seed, width, height, lora_scale):
     if negative_prompt == "":
         negative_prompt = None
     if selected_index is None:
@@ -157,7 +156,7 @@ def run_lora(prompt, negative_prompt, cfg_scale, steps, selected_index, seed, wi
             pipe.load_lora_weights("ostris/OpenFLUX.1", weight_name="openflux1-v0.1.0-fast-lora.safetensors", adapter_weights=[1.0])
             pipe.load_lora_weights(lora_path, adapter_name=selected_lora["repo"], adapter_weights=lora_scale)
     
-    image = generate_image(prompt, negative_prompt, steps, seed, cfg_scale, width, height, lora_scale, progress)
+    image = generate_image(prompt, negative_prompt, steps, seed, cfg_scale, width, height, lora_scale)
     pipe.to("cpu")
     pipe.unload_lora_weights()
     return image, seed