feat: keybert

keyExtractor/keyBert.py

@@ -5,8 +5,12 @@ class KeyExtractor:
         self.model = KeyBERT()
 
     def get_keywords(self, text):
-      keywords = self.model.extract_keywords(text, keyphrase_ngram_range=(1, 3), stop_words=None)
-      return keywords
+      out = []
+      results = self.model.extract_keywords(text, keyphrase_ngram_range=(1, 3), stop_words=None)
+      for result in results:
+        keyword, _ = result
+        out.append(keyword)
+      return out
 
 #import gradio as gr
 #from fastapi import FastAPI