Spaces:
Runtime error
Runtime error
domenicrosati
commited on
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
|