Update app.py

app.py

@@ -1,7 +1,8 @@
 from transformers import AutoProcessor, AutoModelForZeroShotObjectDetection
 import torch
 from transformers import SamModel, SamProcessor
-
+import spaces
+import numpy as np
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 sam_model = SamModel.from_pretrained("facebook/sam-vit-base").to("cuda")
@@ -32,7 +33,7 @@ def infer_dino(img, text_queries, score_threshold):
   return results
 
 
-import numpy as np
+@spaces.GPU
 def query_image(img, text_queries, dino_threshold):
   text_queries = text_queries
   text_queries = text_queries.split(",")
@@ -76,6 +77,5 @@ demo = gr.Interface(
     examples=[
         ["./cats.png", "cat, fishnet", 0.16],["./bee.jpg", "bee, flower", 0.16]
     ],
-    cache_examples=True
 )
 demo.launch(debug=True)