Spaces:
Runtime error
Runtime error
Commit
Β·
2d39184
1
Parent(s):
29232c3
ship smaller model
Browse files
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-
|
| 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
|