Spaces:
Runtime error
Runtime error
update app.py
Browse files
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)
|