Add ZeroGPU support

app.py

@@ -5,6 +5,7 @@ import torch
 import gradio as gr
 import numpy as np
 import imageio
+import spaces
 from PIL import Image
 
 # Add project root to path
@@ -98,6 +99,7 @@ def load_video_frames(video_path: str, source_frames: int):
     return input_video, original_height, original_width
 
 class VideoCoF_Controller(Wan_Controller):
+    @spaces.GPU(duration=240)
     @timer
     def generate(
         self,
@@ -142,7 +144,14 @@ class VideoCoF_Controller(Wan_Controller):
     ):
         self.clear_cache()
         print(f"VideoCoF Generation started.")
-
+        
+        # Ensure model is on CUDA inside the zero-gpu decorated function
+        if torch.cuda.is_available():
+            self.device = torch.device("cuda")
+            # If pipeline is not on cuda, move it (if possible, but usually accelerate handles this or it's handled by parts)
+            # However, Wan_Controller logic might rely on `self.device`.
+            # We explicitly set `self.device` to cuda here.
+        
         if self.diffusion_transformer_dropdown != diffusion_transformer_dropdown:
             self.update_diffusion_transformer(diffusion_transformer_dropdown)