update space

app.py

@@ -9,6 +9,7 @@ import uuid
 import GPUtil
 import gradio as gr
 import psutil
+import spaces
 import torch
 
 from videosys import CogVideoXConfig, CogVideoXPABConfig, VideoSysEngine
@@ -56,12 +57,14 @@ def get_server_status():
     return {"cpu": f"{cpu_percent}%", "memory": f"{memory.percent}%", "disk": f"{disk.percent}%", "gpu": gpu_info}
 
 
+@spaces.GPU
 def generate_vanilla(prompt, num_inference_steps, guidance_scale, progress=gr.Progress(track_tqdm=True)):
     engine = load_model()
     video_path = generate(engine, prompt, num_inference_steps, guidance_scale)
     return video_path
 
 
+@spaces.GPU
 def generate_vs(
     prompt,
     num_inference_steps,