add spaces

app.py

@@ -11,6 +11,7 @@ import torch
 import torchvision
 from pytorch_lightning import seed_everything
 from huggingface_hub import hf_hub_download
+import spaces
 
 sys.path.insert(0, "scripts/evaluation")
 from funcs import (
@@ -39,6 +40,7 @@ def check_move(trajectory, video_length=16):
 
     return True
 
+@spaces.GPU(duration=120)
 def infer(*user_args):
     prompt_in = user_args[0]
     target_indices = user_args[1]