Update app.py

app.py

@@ -2,6 +2,7 @@ import torch
 import gradio as gr
 from diffusers import StableVideoDiffusionPipeline
 from diffusers.utils import load_image, export_to_video
+import spaces
 
 # Check if GPU is available
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -12,6 +13,7 @@ pipeline = StableVideoDiffusionPipeline.from_pretrained(
 )
 pipeline.to(device)
 
+@spaces.GPU
 def generate_video(image_path, seed):
     # Load and preprocess the image
     image = load_image(image_path)