updated testes

Signed-off-by: Prabod Rathnayaka <prabod@rathnayaka.me>
JohnSnowLabs · Nov 8, 2024 · f7e6d2d · f7e6d2d
1 parent bfaf0ec
commit f7e6d2d
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/test/scala/com/johnsnowlabs/nlp/annotators/cv/Phi3VisionTestSpec.scala b/src/test/scala/com/johnsnowlabs/nlp/annotators/cv/Phi3VisionTestSpec.scala
@@ -27,9 +27,9 @@ import org.scalatest.flatspec.AnyFlatSpec
 
 class Phi3VisionTestSpec extends AnyFlatSpec {
 
-  lazy val model = getBLIPForQuestionAnsweringPipelineModel
+  lazy val model = getPhi3VisionPipelineModel
 
-  "BLIP" should "answer a question for a given image" taggedAs SlowTest in {
+  "Phi3Vision" should "answer a question for a given image" taggedAs SlowTest in {
 
     val testDF = getTestDF
     val result = model.transform(testDF)
@@ -152,7 +152,7 @@ class Phi3VisionTestSpec extends AnyFlatSpec {
 
   }
 
-  private def getBLIPForQuestionAnsweringPipelineModel = {
+  private def getPhi3VisionPipelineModel = {
     val testDF = getTestDF
 
     val imageAssembler: ImageAssembler = new ImageAssembler()
@@ -163,6 +163,7 @@ class Phi3VisionTestSpec extends AnyFlatSpec {
       .pretrained()
       .setInputCols("image_assembler")
       .setOutputCol("answer")
+      .setMaxOutputLength(50)
 
     val newPipeline: Pipeline =
       new Pipeline().setStages(Array(imageAssembler, loadModel))