frogleo commited on
Commit
17579c2
·
verified ·
1 Parent(s): 67a28bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -60,8 +60,8 @@ def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5
60
 
61
  def callback_fn(pipe, step, timestep, callback_kwargs):
62
  print(f"[Step {step}] Timestep: {timestep}")
63
- progress_value = (step+1.0)/num_inference_steps
64
- progress(progress_value, desc=f"Image generating, {step + 1}/{num_inference_steps} steps")
65
 
66
  if randomize_seed:
67
  seed = random.randint(0, MAX_SEED)
 
60
 
61
  def callback_fn(pipe, step, timestep, callback_kwargs):
62
  print(f"[Step {step}] Timestep: {timestep}")
63
+ progress_value = (step+1.0)/steps
64
+ progress(progress_value, desc=f"Image generating, {step + 1}/{steps} steps")
65
 
66
  if randomize_seed:
67
  seed = random.randint(0, MAX_SEED)