[SPARKNLP-1011] Adding addFile changes to UAEEmbeddings

JohnSnowLabs · Apr 16, 2024 · be70be8 · be70be8
1 parent e849215
commit be70be8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/scala/com/johnsnowlabs/nlp/embeddings/UAEEmbeddings.scala b/src/main/scala/com/johnsnowlabs/nlp/embeddings/UAEEmbeddings.scala
@@ -515,7 +515,8 @@ trait ReadUAEDLModel extends ReadTensorflowModel with ReadOnnxModel {
           .setModelIfNotSet(spark, Some(wrapper), None)
 
       case ONNX.name =>
-        val onnxWrapper = OnnxWrapper.read(localModelPath, zipped = false, useBundle = true)
+        val onnxWrapper =
+          OnnxWrapper.read(spark, localModelPath, zipped = false, useBundle = true)
         annotatorModel
           .setModelIfNotSet(spark, None, Some(onnxWrapper))