fix spaces

demo/infer.py

@@ -5,6 +5,8 @@ from transformers import Qwen2VLForConditionalGeneration, AutoProcessor, LogitsP
 from livecc_utils import prepare_multiturn_multimodal_inputs_for_generation, get_smart_resized_clip, get_smart_resized_video_reader
 from qwen_vl_utils import process_vision_info
 
+import spaces
+
 logger = logging.get_logger(__name__)
 
 class ThresholdLogitsProcessor(LogitsProcessor):
@@ -31,6 +33,8 @@ class LiveCCDemoInfer:
     initial_time_interval = initial_fps_frames / fps
     streaming_time_interval = streaming_fps_frames / fps
     frame_time_interval = 1 / fps
+
+    @spaces.GPU
     def __init__(self, model_path: str = None, device_id: int = 0):
         self.model = Qwen2VLForConditionalGeneration.from_pretrained(
             model_path, torch_dtype="auto", 
@@ -51,6 +55,7 @@ class LiveCCDemoInfer:
         self.system_prompt_offset = texts.index('<|im_start|>user')
         self._cached_video_readers_with_hw = {}
 
+    @spaces.GPU
     @torch.inference_mode()
     def live_cc(
         self,