change input type

3 years ago · 1162267a6a
1 changed files with 1 additions and 1 deletions
--- 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