Update app_gradio.py

app_gradio.py

@@ -67,7 +67,7 @@ def prepare_latents(pipe, x_aug):
 
 @torch.no_grad()
 def invert(pipe, inv, load_name, device="cuda", dtype=torch.bfloat16):
-    input_img = [load_image(load_name, 256).to(device, dtype=dtype).unsqueeze(1)]
+    input_img = [load_image(load_name, 256).to(device, dtype=dtype).unsqueeze(1)] * 5
     input_img = torch.cat(input_img, dim=1)
     latents = prepare_latents(pipe, input_img).to(torch.bfloat16)
     inv.set_timesteps(25)
@@ -191,7 +191,7 @@ def generate_output(image, apply_filter, prompt: str, num_seeds: int = 3, lambda
     temp_image_path = os.path.join(exp_dir, f"temp_input_{unique_id}.png")
 
     image = Image.open(image)
-    image = image.resize((256, 256), Image.LANCZOS)
+    image = image.resize((256, 256), Image.Resampling.LANCZOS)
 
     image.save(temp_image_path)