Update app.py

app.py

@@ -1,8 +1,10 @@
 import streamlit as st
 import pandas as pd
 from transformers import AutoTokenizer, AutoModel
+# Use a pipeline as a high-level helper
+from transformers import pipeline
 
-# Load the PubMedBERT tokenizer and model directly
+pipe = pipeline("fill-mask", model="microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext")# Load the PubMedBERT tokenizer and model directly
 tokenizer = AutoTokenizer.from_pretrained("microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext")
 model = AutoModel.from_pretrained("microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext")