Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from diffusers import StableVideoDiffusionPipeline, EulerDiscreteScheduler
 import torch
+import GPU
 
 # Load the Stable Video Diffusion model
 model_id = "stabilityai/stable-video-diffusion-img2vid-xt"
@@ -8,6 +9,7 @@ pipe = StableVideoDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.
 pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
 pipe.to("cuda")
 
+@spaces.GPU
 def generate_video(image, prompt, num_frames=25, resolution=(576, 1024)):
     # Generate the video
     video_frames = pipe(prompt, image=image, num_frames=num_frames, height=resolution[0], width=resolution[1]).frames