safinal commited on
Commit
ef08130
Β·
verified Β·
1 Parent(s): 38b5db9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
- "trained_distil_bert_base",
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