Spaces:
Sleeping
Sleeping
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)
|