Update app.py

app.py

@@ -4,6 +4,7 @@ import torch
 from PIL import Image
 import tempfile
 import imageio
+import spaces
 
 # Load the Stable Video Diffusion model
 model_id = "stabilityai/stable-video-diffusion-img2vid-xt"
@@ -14,6 +15,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