Update app.py

app.py

@@ -2,6 +2,7 @@ from transformers import BitsAndBytesConfig, LlavaNextVideoForConditionalGenerat
 import torch
 import numpy as np
 import av
+import spaces
 
 
 quantization_config = BitsAndBytesConfig(
@@ -18,7 +19,7 @@ model = LlavaNextVideoForConditionalGeneration.from_pretrained(
     device_map='auto'
 )
 
-
+@spaces.GPU
 def read_video_pyav(container, indices):
     '''
     Decode the video with PyAV decoder.
@@ -41,6 +42,7 @@ def read_video_pyav(container, indices):
             frames.append(frame)
     return np.stack([x.to_ndarray(format="rgb24") for x in frames])
 
+@spaces.GPU
 def process_video(video_file, question):
     # Open video and sample frames
     with av.open(video_file) as container: