add zero

app.py

@@ -10,6 +10,7 @@ import torch
 import numpy as np
 from PIL import Image
 from transformers import AutoImageProcessor, AutoModelForObjectDetection
+import spaces
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
@@ -71,7 +72,7 @@ def process_video(
             )
             sink.write_frame(frame)
     return result_file_path
-
+@spaces.GPU
 def query(image):
   inputs = processor(images=image, return_tensors="pt").to(device)
   with torch.no_grad():