Update injection_main.py

injection_main.py

@@ -327,8 +327,10 @@ def invert(
     # Reversed timesteps <<<<<<<<<<<<<<<<<<<<
     timesteps = list(reversed(pipe.scheduler.timesteps))
 
-    for i in tqdm(range(1, num_inference_steps), total=num_inference_steps - 1, desc="DDIM Inversion"):
+    # Initialize tqdm progress bar
+    progress_bar = tqdm(range(1, num_inference_steps), total=num_inference_steps - 1, desc="DDIM Inversion")
 
+    for i in progress_bar:
         # We'll skip the final iteration
         if i >= num_inference_steps - 1:
             continue
@@ -373,6 +375,7 @@ def invert(
 
 
 
+
 def style_image_with_inversion(
     pipe,
     input_image,