Undo keybert removal

models.py

@@ -28,19 +28,19 @@ def create_nest_sentences(document:str, token_max_length = 1024):
   return nested
 
 # Reference: https://github.com/MaartenGr/KeyBERT
-# @st.cache(allow_output_mutation=True)
-# def load_keyword_model():
-#   kw_model = KeyBERT()
-#   return kw_model
-
-# def keyword_gen(kw_model, sequence:str):
-#   keywords = kw_model.extract_keywords(sequence, 
-#     keyphrase_ngram_range=(1, 1),
-#     stop_words='english', 
-#     use_mmr=True, 
-#     diversity=0.5,
-#     top_n=10)
-#   return keywords
+@st.cache(allow_output_mutation=True)
+def load_keyword_model():
+  kw_model = KeyBERT()
+  return kw_model
+
+def keyword_gen(kw_model, sequence:str):
+  keywords = kw_model.extract_keywords(sequence, 
+    keyphrase_ngram_range=(1, 1),
+    stop_words='english', 
+    use_mmr=True, 
+    diversity=0.5,
+    top_n=10)
+  return keywords