Update app.py

app.py

@@ -9,6 +9,7 @@ import requests
 import torch
 import numpy as np
 import gradio as gr
+import spaces
 
 ## NLLB Inference
 nllb_clip_processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
@@ -73,6 +74,7 @@ def infer_siglip(image, candidate_labels):
   siglip_out = siglip_detector(image, candidate_labels)
   return postprocess_siglip(siglip_out, labels=candidate_labels)
 
+@spaces.GPU
 def infer(image, labels):
   st_out = infer_st(image, labels)
   nllb_out = nllb_clip_inference(image, labels)