MaartenGr / KeyBERT

Minimal keyword extraction with BERT

Home Page:https://MaartenGr.github.io/KeyBERT/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Using KeyBERT with a locally saved model

ppsoong opened this issue · comments

Is it possible to use KeyBERT with a pre-trained model I have saved locally? I didn't see anything in the documentation about it.

I'm going to close this. After further review it seems like I should be able to load the local model using SentenceTransformer() or Huggingface's pipeline() and load that into KeyBERT.