Update app.py

app.py

@@ -45,7 +45,13 @@ pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 MAX_SEED = np.iinfo(np.int32).max
 
 @spaces.GPU
-def run(prompt="a photo of an astronaut riding a horse on mars", steps=10, seed=20, negative_prompt="", randomize_seed=False):
+def run(prompt="a photo of an astronaut riding a horse on mars",
+        steps=10,
+        seed=20,
+        negative_prompt="",
+        randomize_seed=False,
+        progress=gr.Progress(track_tqdm=True)
+    ):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)