diff --git a/efficientnet_image_embedding.yaml b/efficientnet_image_embedding.yaml
index 33afc2d..77efa95 100644
--- a/efficientnet_image_embedding.yaml
+++ b/efficientnet_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