renaming tf file name in reading

JohnSnowLabs · Mar 16, 2024 · 7e5d33e · 7e5d33e
1 parent a3accf8
commit 7e5d33e
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/.../scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForQuestionAnswering.scala b/.../scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForQuestionAnswering.scala
@@ -319,7 +319,7 @@ trait ReadXlmRoBertaForQuestionAnsweringDLModel
     instance.getEngine match {
       case TensorFlow.name =>
         val tfWrapper =
-          readTensorflowModel(path, spark, "xlm_roberta_classification_tf", initAllTables = false)
+          readTensorflowModel(path, spark, "xlm_roberta_classification_tensorflow", initAllTables = false)
         instance.setModelIfNotSet(spark, Some(tfWrapper), None, spp)
       case ONNX.name =>
         val onnxWrapper =

diff --git a/...a/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForSequenceClassification.scala b/...a/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForSequenceClassification.scala
@@ -370,7 +370,7 @@ trait ReadXlmRoBertaForSequenceDLModel
     instance.getEngine match {
       case TensorFlow.name =>
         val tfWrapper =
-          readTensorflowModel(path, spark, "xlm_roberta_classification_tf", initAllTables = false)
+          readTensorflowModel(path, spark, "xlm_roberta_classification_tensorflow", initAllTables = false)
         instance.setModelIfNotSet(spark, Some(tfWrapper), None, spp)
       case ONNX.name =>
         val onnxWrapper =

diff --git a/...cala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForTokenClassification.scala b/...cala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForTokenClassification.scala
@@ -342,7 +342,7 @@ trait ReadXlmRoBertaForTokenDLModel
     instance.getEngine match {
       case TensorFlow.name =>
         val tfWrapper =
-          readTensorflowModel(path, spark, "xlm_roberta_classification_tf", initAllTables = false)
+          readTensorflowModel(path, spark, "xlm_roberta_classification_tensorflow", initAllTables = false)
         instance.setModelIfNotSet(spark, Some(tfWrapper), None, spp)
       case ONNX.name =>
         val onnxWrapper =