Hugging Face's logo

Spaces:
pritamdeka
/
Biomedical-Fact-Checker
Sleeping

App Files Community
pritamdeka commited on
Commit
7e2f865
·
verified ·
1 Parent(s): 3a7eaa2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ import torch
18
 
19
  # --- Models (load once, globally) ---
20
  scispacy = en_core_sci_lg.load()
21
- sbert_keybert = SentenceTransformer("pritamdeka/PubMedBERT-mnli-snli-scinli-scitail-mednli-stsb")
22
  sbert_rerank = SentenceTransformer("pritamdeka/S-PubMedBert-MS-MARCO")
23
  NLI_MODEL_NAME = "pritamdeka/PubMedBERT-MNLI-MedNLI"
24
  nli_tokenizer = AutoTokenizer.from_pretrained(NLI_MODEL_NAME)
 
18
 
19
  # --- Models (load once, globally) ---
20
  scispacy = en_core_sci_lg.load()
21
+ sbert_keybert = SentenceTransformer("pritamdeka/BioBERT-mnli-snli-scinli-scitail-mednli-stsb")
22
  sbert_rerank = SentenceTransformer("pritamdeka/S-PubMedBert-MS-MARCO")
23
  NLI_MODEL_NAME = "pritamdeka/PubMedBERT-MNLI-MedNLI"
24
  nli_tokenizer = AutoTokenizer.from_pretrained(NLI_MODEL_NAME)