Fix face landmark model URL and adjust model resource check

Signed-off-by: vinlemon <[email protected]>
WasmEdge · Jul 17, 2024 · fd37177 · fd37177
1 parent 06464f4
commit fd37177
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/scripts/download-models.sh b/scripts/download-models.sh
@@ -131,7 +131,7 @@ face_landmark_init() {
     mkdir -p "${face_landmark_dir}"
     pushd "${face_landmark_dir}"
 
-    model_urls=("https://storage.googleapis.com/mediapipe-assets/face_landmark.tflite"
+    model_urls=("https://storage.googleapis.com/mediapipe-tasks/face_landmarker/face_landmarker.task"
     )
 
     for url in "${model_urls[@]}"; do

diff --git a/src/tasks/vision/face_landmark/builder.rs b/src/tasks/vision/face_landmark/builder.rs
@@ -72,7 +72,7 @@ impl FaceLandmarkerBuilder {
         let model_resource = crate::model::parse_model(landmark_file.as_ref())?;
 
         // check model
-        model_base_check_impl!(model_resource, 1, 3);
+        model_base_check_impl!(model_resource, 1, 2);
         model_resource_check_and_get_impl!(model_resource, to_tensor_info, 0).try_to_image()?;
         let input_tensor_type =
             model_resource_check_and_get_impl!(model_resource, input_tensor_type, 0);