Update 2023-11-10-legembeddings_bge_base_en.md

JohnSnowLabs · Dec 13, 2023 · fac354a · fac354a
1 parent c04998a
commit fac354a
Showing 1 changed file with 2 additions and 2 deletions.
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
@@ -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(