加入进度支持

app.py

@@ -58,8 +58,8 @@ def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5
 
     def callback_fn(step: int, timestep: int, latents: torch.FloatTensor):
         print(f"[Step {step}] Timestep: {timestep}")
-        # progress_value = (step+1.0)/num_inference_steps
-        # progress(progress_value, desc=f"Image generating, {step + 1}/{num_inference_steps} steps")
+        progress_value = (step+1.0)/num_inference_steps
+        progress(progress_value, desc=f"Image generating, {step + 1}/{num_inference_steps} steps")
 
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)