Update app.py

app.py

@@ -1,6 +1,7 @@
 import torch
 
 from keybert import KeyBERT
+from sentence_transformers import SentenceTransformer
 from keyphrase_vectorizers import KeyphraseCountVectorizer
 from transformers import T5ForConditionalGeneration,T5Tokenizer
 
@@ -17,7 +18,8 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 HfFolder.save_token(st.secrets["hf-auth-token"])
 
 # Load KeyBert Model
-kw_extractor = KeyBERT('valurank/MiniLM-L6-Keyword-Extraction', use_auth_token=True)
+tmp_model = SentenceTransformer('valurank/MiniLM-L6-Keyword-Extraction', use_auth_token=True)
+kw_extractor = KeyBERT(tmp_model)
 
 # Load T5 for Paraphrasing
 t5_model = T5ForConditionalGeneration.from_pretrained('valurank/t5-paraphraser', use_auth_token=True)