WalidBouss commited on
Commit
ff11be0
·
verified ·
1 Parent(s): bb93040

update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -55,7 +55,7 @@ def logits_to_heatmaps(logits, image_cv):
55
 
56
 
57
  # ---------- Main visualization function ----------
58
- @spaces.GPU
59
  def viz_func(url, image, text_query):
60
  image_torch = preprocess(image).unsqueeze(0).to(device)
61
  text_emb = _get_text_embedding(model, tokenizer, classes=[text_query], device=device)
 
55
 
56
 
57
  # ---------- Main visualization function ----------
58
+ # @spaces.GPU
59
  def viz_func(url, image, text_query):
60
  image_torch = preprocess(image).unsqueeze(0).to(device)
61
  text_emb = _get_text_embedding(model, tokenizer, classes=[text_query], device=device)