Update app.py

app.py

@@ -19,9 +19,8 @@ class WanAnimateApp:
             model_name,
             torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
             variant="fp16",
-            device_map="cpu"  # With accelerate, this works
+            device_map="cpu"
         )
-        self.pipe.enable_model_cpu_offload()  # Now safe with accelerate
 
     def predict(
         self, 
@@ -34,8 +33,12 @@ class WanAnimateApp:
             return None, "Upload both image and video."
 
         try:
-            # Local processing
-            ref_image = Image.open(ref_img).convert("RGB").resize((576, 320))
+            # Local processing — PIL directly (no open for PIL type)
+            if isinstance(ref_img, Image.Image):
+                ref_image = ref_img.convert("RGB").resize((576, 320))
+            else:
+                ref_image = Image.open(ref_img).convert("RGB").resize((576, 320))
+
             cap = cv2.VideoCapture(video)
             frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
             cap.release()
@@ -142,7 +145,7 @@ def start_app():
             with gr.Column():    
                 ref_img = gr.Image(
                     label="Reference Image(参考图像)",
-                    type="filepath",
+                    type="pil",  # PIL type to fix FileNotFound
                     sources=["upload"],
                 )