Update app.py

app.py

@@ -19,8 +19,8 @@ kp_cosine_finbert_checkpoint = "cosine_kp_finance.pickle"
 
 tokenizer_sapbert = AutoTokenizer.from_pretrained("cambridgeltl/SapBERT-from-PubMedBERT-fulltext")
 model_sapbert = AutoModel.from_pretrained("cambridgeltl/SapBERT-from-PubMedBERT-fulltext", output_hidden_states=True)
-kp_dict_sapbert_checkpoint = "kp_dict_sapbert.pickle"
-kp_cosine_sapbert_checkpoint = "cosine_kp_sapbert.pickle"
+kp_dict_sapbert_checkpoint = "kp_dict_medical.pickle"
+kp_cosine_sapbert_checkpoint = "cosine_kp_medical.pickle"
 
 text = st.text_input("Enter word or key-phrase")
 exclude_words = st.radio("exclude_words",[True,False], help="Exclude results that contain any words in the query")