Skip to content

Commit 0bb77b9

Browse files
committed
changed to pretrained
Signed-off-by: Prabod Rathnayaka <[email protected]>
1 parent 3a9cae6 commit 0bb77b9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/test/annotator/cv/phi3_vision_for_multimodal_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def setUp(self):
3434

3535
image_assembler = ImageAssembler().setInputCol("image").setOutputCol("image_assembler")
3636

37-
imageClassifier = Phi3Vision.loadSavedModel("/mnt/research/Projects/ModelZoo/Phi-3.5-vision/model/INT4", self.spark) \
37+
imageClassifier = Phi3Vision.pretrained() \
3838
.setInputCols("image_assembler") \
3939
.setOutputCol("answer")
4040

0 commit comments

Comments
 (0)