add @spaces.GPU

util/v3d.py

@@ -5,6 +5,7 @@ import torch
 from torchvision.utils import make_grid
 import tempfile
 import gradio as gr
+import spaces
 from omegaconf import OmegaConf
 from einops import rearrange
 from scripts.pub.V3D_512 import (
@@ -25,7 +26,7 @@ from glob import glob
 from mediapy import write_video
 from pathlib import Path
 
-
+@spaces.GPU
 def generate_v3d(
     image,
     model,