change input type

4 years ago · e1161a7263
1 changed files with 1 additions and 1 deletions
--- 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