update space

app.py

@@ -6,6 +6,7 @@ from PIL import Image
 import numpy as np
 import transformers
 from typing import Dict, Optional, Sequence, List
+import spaces
 
 import sys
 from oryx.conversation import conv_templates, SeparatorStyle
@@ -77,7 +78,7 @@ def preprocess_qwen(sources, tokenizer: transformers.PreTrainedTokenizer, has_im
     targets = torch.tensor(targets, dtype=torch.long)
     return input_ids
 
-
+@spaces.GPU(duration=120)
 def oryx_inference(video, text):
     vr = VideoReader(video, ctx=cpu(0))
     total_frame_num = len(vr)