Update app.py

app.py

@@ -4,6 +4,7 @@ import imageio
 from diffusers import StableVideoDiffusionPipeline
 from diffusers.utils import load_image, export_to_video
 import gradio as gr
+import spaces
 
 # Load the pipeline
 pipe = StableVideoDiffusionPipeline.from_pretrained(
@@ -14,6 +15,7 @@ pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
 pipe.enable_model_cpu_offload()
 pipe.unet.enable_forward_chunking()
 
+@spaces.GPU
 def generate_video(image, seed=42, fps=7, motion_bucket_id=180, noise_aug_strength=0.1):
     # Resize the image
     image = image.resize((1024, 576))