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):
+def generate_image(prompt, negative_prompt, trigger_word, steps, seed, cfg_scale, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
     pipe.to("cuda")
     generator = torch.Generator(device="cuda").manual_seed(seed)
     
@@ -123,6 +123,7 @@ 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