diff --git a/resnet_image_embedding.yaml b/resnet_image_embedding.yaml
index ec63089..419b6fe 100644
--- a/resnet_image_embedding.yaml
+++ b/resnet_image_embedding.yaml
@@ -8,6 +8,6 @@ init:
   model_name: str
 call:
   input:
-    img_tensor: torch.Tensor
+    img_path: str
   output:
     feature_vector: numpy.ndarray