Update app.py

app.py

@@ -3,7 +3,7 @@ from diffusers import StableVideoDiffusionPipeline, EulerDiscreteScheduler
 import torch
 from PIL import Image
 import tempfile
-
+import spaces
 # Load the Stable Video Diffusion model
 model_id = "stabilityai/stable-video-diffusion-img2vid-xt"
 try:
@@ -13,6 +13,7 @@ try:
 except Exception as e:
     raise RuntimeError(f"Failed to load the model: {e}")
 
+@spaces.GPU
 def generate_video(image, num_frames=25, height=576, width=1024):
     try:
         # Convert the image to a format suitable for the pipeline