Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -75,7 +75,7 @@ def process_single_query(model, query_image_path, query_text, database_embedding
|
|
| 75 |
|
| 76 |
# Get token classifier
|
| 77 |
token_classifier, token_classifier_tokenizer = load_token_classifier(
|
| 78 |
-
"
|
| 79 |
device
|
| 80 |
)
|
| 81 |
|
|
|
|
| 75 |
|
| 76 |
# Get token classifier
|
| 77 |
token_classifier, token_classifier_tokenizer = load_token_classifier(
|
| 78 |
+
"safinal/compositional-retrieval-token-classifier",
|
| 79 |
device
|
| 80 |
)
|
| 81 |
|