add support for progress bar

app.py

@@ -97,6 +97,7 @@ def call_model(
         ema: bool = True,
         steps: int = 1_000,
         num_images: int = 0,
+        progress=gr.Progress(track_tqdm=True),
 ):
     diffusion_model = linear_diffusion_model if model_to_call.lower() == "linear" else cosine_diffusion_model
     return diffusion_model.generate_images(