Update app.py

app.py

@@ -3,14 +3,13 @@ import torch
 import spaces
 from typing import List
 from PIL import Image
-from diffusers import AutoencoderKLWan, LucyEditPipeline
+from diffusers import LucyEditPipeline
 from diffusers.utils import export_to_video, load_video
 import tempfile
 import os
 
 model_id = "decart-ai/Lucy-Edit-Dev"
-vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
-pipe = LucyEditPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
+pipe = LucyEditPipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16)
 pipe.to("cuda")
 
 def calculate_resolution(input_width, input_height, max_dimension=832):