diff --git a/docs/_posts/gadde5300/2023-11-10-legembeddings_bge_base_en.md b/docs/_posts/gadde5300/2023-11-10-legembeddings_bge_base_en.md index 1c9d27a8ba..c493d26458 100644 --- a/docs/_posts/gadde5300/2023-11-10-legembeddings_bge_base_en.md +++ b/docs/_posts/gadde5300/2023-11-10-legembeddings_bge_base_en.md @@ -46,8 +46,8 @@ tokenizer = nlp.Tokenizer() \ .setInputCols("document") \ .setOutputCol("token") -BGE_loaded = nlp.BertEmbeddings.load("legembeddings_bge_base","en", "legal/models")\ - .setInputCols(["document","token"])\ +BGE_loaded = nlp.BertEmbeddings.pretrained("legembeddings_bge_base", "en", "legal/models")\ + .setInputCols(["document", "token"])\ .setOutputCol("BGE")\ pipeline = nlp.Pipeline(