Spaces:
Runtime error
Runtime error
Commit
Β·
a5a19f3
1
Parent(s):
964c419
update cross encoder
Browse files
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-
|
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
|