Update app.py

app.py

@@ -19,8 +19,9 @@ class WanAnimateApp:
             model_name,
             torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
             variant="fp16",
-            device_map="cpu"
+            device_map="cpu"  # With accelerate, this works
         )
+        self.pipe.enable_model_cpu_offload()  # Now safe with accelerate
 
     def predict(
         self, 
@@ -33,18 +34,8 @@ class WanAnimateApp:
             return None, "Upload both image and video."
 
         try:
-            # Fix Gradio temp file: Always save to real temp and open
-            with tempfile.NamedTemporaryFile(suffix='.png', delete=False) as tmp:
-                if isinstance(ref_img, Image.Image):
-                    ref_img.save(tmp.name)
-                else:
-                    # If bytes or str, handle
-                    if isinstance(ref_img, bytes):
-                        ref_img = Image.open(io.BytesIO(ref_img))
-                    ref_img.save(tmp.name)
-                ref_image = Image.open(tmp.name).convert("RGB").resize((576, 320))
-            os.unlink(tmp.name)  # Clean up
-
+            # Local processing
+            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()