domenicrosati commited on
Commit
f2eab41
Β·
1 Parent(s): d08af75

update cross-encoder

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -79,7 +79,7 @@ def init_models():
79
  "question-answering", model='sultan/BioM-ELECTRA-Large-SQuAD2-BioASQ8B',
80
  device=device
81
  )
82
- reranker = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-6-v2', device=device)
83
  queryexp_tokenizer = AutoTokenizer.from_pretrained("doc2query/all-with_prefix-t5-base-v1")
84
  queryexp_model = AutoModelWithLMHead.from_pretrained("doc2query/all-with_prefix-t5-base-v1")
85
  return question_answerer, reranker, stop, device, queryexp_model, queryexp_tokenizer
 
79
  "question-answering", model='sultan/BioM-ELECTRA-Large-SQuAD2-BioASQ8B',
80
  device=device
81
  )
82
+ reranker = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-12-v2', device=device)
83
  queryexp_tokenizer = AutoTokenizer.from_pretrained("doc2query/all-with_prefix-t5-base-v1")
84
  queryexp_model = AutoModelWithLMHead.from_pretrained("doc2query/all-with_prefix-t5-base-v1")
85
  return question_answerer, reranker, stop, device, queryexp_model, queryexp_tokenizer