Update app.py

app.py

@@ -18,13 +18,8 @@ import sys
 def load_models():
     device = "cuda" if torch.cuda.is_available() else "cpu"
 
-    pubmed_tok = AutoTokenizer.from_pretrained(
-        "microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract"
-    )
-    pubmed_model = AutoModel.from_pretrained(
-        "microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract",
-        use_safetensors=True
-    ).to(device)
+    pubmed_tok = AutoTokenizer.from_pretrained("microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract")
+    pubmed_model = AutoModel.from_pretrained("microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract").to(device)
 
     try:
         nlp = spacy.load("en_core_sci_sm")