Update app.py

app.py

@@ -25,7 +25,7 @@ optimize_pipeline_(
 print("Pipeline loaded!")
 
 @spaces.GPU
-def generate_image(prompt, height, width, num_inference_steps, seed, randomize_seed):
+def generate_image(prompt, height, width, num_inference_steps, seed, randomize_seed, progress=gr.Progress(track_tqdm=True)):
     """Generate an image from the given prompt."""
     if randomize_seed:
         seed = torch.randint(0, 2**32 - 1, (1,)).item()