hysts HF staff commited on
Commit
4d961a4
·
1 Parent(s): f1ef7a2
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -77,6 +77,7 @@ def generate(
77
  num_inference_steps_base: int = 25,
78
  num_inference_steps_refiner: int = 25,
79
  apply_refiner: bool = False,
 
80
  ) -> PIL.Image.Image:
81
  generator = torch.Generator().manual_seed(seed)
82
 
 
77
  num_inference_steps_base: int = 25,
78
  num_inference_steps_refiner: int = 25,
79
  apply_refiner: bool = False,
80
+ progress: gr.Progress = gr.Progress(track_tqdm=True), # noqa: ARG001, B008
81
  ) -> PIL.Image.Image:
82
  generator = torch.Generator().manual_seed(seed)
83