Update injection_main.py

injection_main.py

@@ -328,10 +328,9 @@ def invert(
     timesteps = list(reversed(pipe.scheduler.timesteps))
 
     # Initialize tqdm progress bar
-    progress_bar = tqdm(range(num_inference_steps - 1), total=num_inference_steps - 1, desc="DDIM Inversion")
-
-    for i in progress_bar:
+    for i in range(num_inference_steps - 1):
         print(f"Iteration type: {type(i)}, value: {i}")  # Debugging statement
+
         if isinstance(i, (int, float)):
             i = int(i)
         else:
@@ -384,6 +383,7 @@ def invert(
 
 
 
+
 def style_image_with_inversion(
     pipe,
     input_image,