Remove keybert

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