Update app.py
Browse files
app.py
CHANGED
@@ -16,8 +16,8 @@ import torch
|
|
16 |
|
17 |
# --- Models (load once, globally) ---
|
18 |
scispacy = en_core_sci_lg.load()
|
19 |
-
sbert_keybert = SentenceTransformer("pritamdeka/
|
20 |
-
sbert_rerank = SentenceTransformer("pritamdeka/
|
21 |
NLI_MODEL_NAME = "pritamdeka/PubMedBERT-MNLI-MedNLI"
|
22 |
nli_tokenizer = AutoTokenizer.from_pretrained(NLI_MODEL_NAME)
|
23 |
nli_model = AutoModelForSequenceClassification.from_pretrained(NLI_MODEL_NAME)
|
|
|
16 |
|
17 |
# --- Models (load once, globally) ---
|
18 |
scispacy = en_core_sci_lg.load()
|
19 |
+
sbert_keybert = SentenceTransformer("pritamdeka/BioBERT-mnli-snli-scinli-scitail-mednli-stsb") # for keybert query
|
20 |
+
sbert_rerank = SentenceTransformer("pritamdeka/S-PubMedBert-MS-MARCO") # for abstract reranking
|
21 |
NLI_MODEL_NAME = "pritamdeka/PubMedBERT-MNLI-MedNLI"
|
22 |
nli_tokenizer = AutoTokenizer.from_pretrained(NLI_MODEL_NAME)
|
23 |
nli_model = AutoModelForSequenceClassification.from_pretrained(NLI_MODEL_NAME)
|