Spaces:
Runtime error
Runtime error
domenicrosati
commited on
Commit
Β·
f2eab41
1
Parent(s):
d08af75
update cross-encoder
Browse files
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-
|
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
|