Update app.py

app.py

@@ -13,8 +13,7 @@ pipe = CogVideoXImageToVideoPipeline.from_pretrained(
 def generate_video(prompt, image):
     # Ensure the generator is on the same device as the model
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-    pipe.vae.enable_tiling()
-    pipe.vae.enable_slicing()
+    pipe.to("cuda")
     generator = torch.Generator(device=device).manual_seed(42)
     
     video = pipe(