Update app.py

app.py

@@ -1,7 +1,7 @@
 import torch
 import gradio as gr
 from transformers import Owlv2Processor, Owlv2ForObjectDetection
-
+import spaces
 
 # Use GPU if available
 if torch.cuda.is_available():
@@ -12,7 +12,7 @@ else:
 model = Owlv2ForObjectDetection.from_pretrained("google/owlv2-base-patch16-ensemble").to(device)
 processor = Owlv2Processor.from_pretrained("google/owlv2-base-patch16-ensemble")
 
-
+@spaces.GPU
 def query_image(img, text_queries, score_threshold):
     text_queries = text_queries
     text_queries = text_queries.split(",")