clementchadebec commited on
Commit
38047c4
1 Parent(s): 592c899

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -64,7 +64,7 @@ NUM_INFERENCE_STEPS = 4
64
 
65
 
66
  @spaces.GPU
67
- def infer(prompt, seed, randomize_seed, guidance_scale, num_inference_steps, negative_prompt):
68
  if randomize_seed:
69
  seed = random.randint(0, MAX_SEED)
70
 
 
64
 
65
 
66
  @spaces.GPU
67
+ def infer(prompt, seed, randomize_seed, guidance_scale, num_inference_steps, negative_prompt, progress=gr.Progress(track_tqdm=True)):
68
  if randomize_seed:
69
  seed = random.randint(0, MAX_SEED)
70