Skip to content

Instantly share code, notes, and snippets.

@robgon-art

robgon-art/get_keywords.py

Created May 7, 2022 17:41
Show Gist options
  • Star 1 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save robgon-art/7d438f3fcb3d012af594bfd7e2313549 to your computer and use it in GitHub Desktop.
Save robgon-art/7d438f3fcb3d012af594bfd7e2313549 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
get_keywords.py
from keybert import KeyBERT
kw_model = KeyBERT()
summary = "After leaving her fiancé, Barry, at the altar, Rachel finds herself in Central Perk Café, soaking wet in her wedding dress."
theme = kw_model.extract_keywords(summary, keyphrase_ngram_range=(1, 3))[0][0]
print(theme)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment