KingNish Walmart-the-bag commited on
Commit
ab38e80
1 Parent(s): 15868bd

add status bar so people can see eta (#1)

Browse files

- add status bar so people can see eta (05f94c6bbf03bf45281dd1c91fefa4676799bf10)


Co-authored-by: wbag <[email protected]>

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ MAX_SEED = np.iinfo(np.int32).max
17
  MAX_IMAGE_SIZE = 4096
18
 
19
  @spaces.GPU(duration=20,queue=False)
20
- def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
21
  if randomize_seed:
22
  seed = random.randint(0, MAX_SEED)
23
  generator = torch.Generator().manual_seed(seed)
 
17
  MAX_IMAGE_SIZE = 4096
18
 
19
  @spaces.GPU(duration=20,queue=False)
20
+ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, progress=gr.Progress(track_tqdm=True)):
21
  if randomize_seed:
22
  seed = random.randint(0, MAX_SEED)
23
  generator = torch.Generator().manual_seed(seed)