diff --git a/resnet_image_embedding.yaml b/resnet_image_embedding.yaml
index 419b6fe..72d0952 100644
--- a/resnet_image_embedding.yaml
+++ b/resnet_image_embedding.yaml
@@ -8,6 +8,6 @@ init:
   model_name: str
 call:
   input:
-    img_path: str
+    image: towhee.types.Image
   output:
     feature_vector: numpy.ndarray