domenicrosati commited on
Commit
2d39184
Β·
1 Parent(s): 29232c3

ship smaller model

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