adding BGEEmbeddings to resource downloader (#14133)

JohnSnowLabs · Jan 18, 2024 · 2b93e02 · 2b93e02
1 parent 818de8d
commit 2b93e02
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/scala/com/johnsnowlabs/nlp/pretrained/ResourceDownloader.scala b/src/main/scala/com/johnsnowlabs/nlp/pretrained/ResourceDownloader.scala
@@ -681,7 +681,8 @@ object PythonResourceDownloader {
     "InstructorEmbeddings" -> InstructorEmbeddings,
     "E5Embeddings" -> E5Embeddings,
     "MPNetEmbeddings" -> MPNetEmbeddings,
-    "CLIPForZeroShotClassification" -> CLIPForZeroShotClassification)
+    "CLIPForZeroShotClassification" -> CLIPForZeroShotClassification,
+    "BGEEmbeddings" -> BGEEmbeddings)
 
   // List pairs of types such as the one with key type can load a pretrained model from the value type
   val typeMapper: Map[String, String] = Map("ZeroShotNerModel" -> "RoBertaForQuestionAnswering")