Update app.py

app.py

@@ -190,6 +190,7 @@ def generate_image(
     num_inference_steps=50,
     shift=3.0,
     max_sequence_length=512,
+    progress=gr.Progress(track_tqdm=True)
 ):
     height, width = get_resolution(resolution)
 
@@ -377,7 +378,7 @@ def prompt_enhance(prompt, enable_enhance):
 
 
 @spaces.GPU
-def generate(prompt, resolution, seed, steps, shift, enhance, random_seed, gallery_images):
+def generate(prompt, resolution, seed, steps, shift, enhance, random_seed, gallery_images, progress=gr.Progress(track_tqdm=True)):
     if pipe is None:
         raise gr.Error("Model not loaded.")