Add zero

app.py

@@ -1,12 +1,14 @@
 from transformers import pipeline, SamModel, SamProcessor
 import torch
 import numpy as np
+import spaces
 
 checkpoint = "google/owlvit-base-patch16"
 detector = pipeline(model=checkpoint, task="zero-shot-object-detection")
 sam_model = SamModel.from_pretrained("facebook/sam-vit-base").to("cuda")
 sam_processor = SamProcessor.from_pretrained("facebook/sam-vit-base")
 
+@spaces.GPU
 def query(image, texts, threshold):
   texts = texts.split(",")
   print(texts)