domenicrosati commited on
Commit
a5a19f3
Β·
1 Parent(s): 964c419

update cross encoder

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -132,7 +132,7 @@ def init_models():
132
  "question-answering", model='sultan/BioM-ELECTRA-Large-SQuAD2-BioASQ8B',
133
  device=device
134
  )
135
- reranker = CrossEncoder('cross-encoder/ms-marco-TinyBERT-L-2-v2', device=device)
136
  # queryexp_tokenizer = AutoTokenizer.from_pretrained("doc2query/all-with_prefix-t5-base-v1")
137
  # queryexp_model = AutoModelWithLMHead.from_pretrained("doc2query/all-with_prefix-t5-base-v1")
138
  return question_answerer, reranker, stop, device # uqeryexp_model, queryexp_tokenizer
 
132
  "question-answering", model='sultan/BioM-ELECTRA-Large-SQuAD2-BioASQ8B',
133
  device=device
134
  )
135
+ reranker = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-6-v2', device=device)
136
  # queryexp_tokenizer = AutoTokenizer.from_pretrained("doc2query/all-with_prefix-t5-base-v1")
137
  # queryexp_model = AutoModelWithLMHead.from_pretrained("doc2query/all-with_prefix-t5-base-v1")
138
  return question_answerer, reranker, stop, device # uqeryexp_model, queryexp_tokenizer