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)
|