Update app.py

app.py

@@ -66,6 +66,8 @@ def infer(prompt, ip_adapter_image, negative_prompt="", seed=42, randomize_seed=
     
     generator = torch.Generator(device="cuda").manual_seed(seed)
     pipe = pipe.to("cuda")
+    image_encoder = image_encoder.to("cuda")
+    pipe.image_encoder = image_encoder
     pipe.set_ip_adapter_scale([ip_adapter_scale])
     
     image = pipe(