pritamdeka commited on
Commit
a26345b
·
verified ·
1 Parent(s): fc63a12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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/S-BioBert-snli-multinli-stsb") # for keybert query
20
- sbert_rerank = SentenceTransformer("pritamdeka/s-pubmedbert-msmarco") # 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)
 
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)