diff --git a/models/nanosam_resnet18_image_encoder_1024x1024/model.yml b/models/nanosam_resnet18_image_encoder_1024x1024/model.yml
index 95f0f12..8cc9109 100644
--- a/models/nanosam_resnet18_image_encoder_1024x1024/model.yml
+++ b/models/nanosam_resnet18_image_encoder_1024x1024/model.yml
@@ -37,4 +37,4 @@ meta:
     url: https://github.com/NVIDIA-AI-IOT/nanosam
   uses_depth: false
   verbose_name: NanoSAM embedder
-task_type: classification
+task_type: feature_extraction