Update app.py

app.py

@@ -8,6 +8,7 @@ import torch.nn.functional as F
 import open_clip
 
 import gradio as gr
+import spaces
 
 from legrad import LeWrapper, LePreprocess
 
@@ -54,6 +55,7 @@ def logits_to_heatmaps(logits, image_cv):
 
 
 # ---------- Main visualization function ----------
+@spaces.GPU
 def viz_func(url, image, text_query):
     image_torch = preprocess(image).unsqueeze(0).to(device)
     text_emb = _get_text_embedding(model, tokenizer, classes=[text_query], device=device)