Update app.py

app.py

@@ -12,6 +12,7 @@ import gradio as gr
 import warnings
 
 import torch
+import spaces
 
 os.system("python setup.py build develop --user")
 os.system("pip install packaging==21.3")
@@ -64,6 +65,7 @@ def image_transform_grounding_for_vis(init_image):
 
 model = load_model_hf(config_file, ckpt_repo_id, ckpt_filenmae)
 
+@spaces.GPU
 def run_grounding(input_image, grounding_caption, box_threshold, text_threshold):
     init_image = input_image.convert("RGB")
     original_size = init_image.size