From bb58aec5772c4769510cbc0390244be226419520 Mon Sep 17 00:00:00 2001
From: shiyu22 <shiyu.chen@zilliz.com>
Date: Thu, 16 Dec 2021 14:20:46 +0800
Subject: [PATCH] Update YAML

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
---
 transform_image_operator_template.yaml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/transform_image_operator_template.yaml b/transform_image_operator_template.yaml
index 13e00eb..835b1c1 100644
--- a/transform_image_operator_template.yaml
+++ b/transform_image_operator_template.yaml
@@ -3,11 +3,11 @@ labels:
   recommended_framework: #the recommend framework, such as 'pytorch1.2.0'
   class: image-transform
   others:
-operator: 'towhee/transform-image-operator-class'
+operator: 'towhee/transform-image-operator-template'
 init:
   size: int
 call:
   input:
-    img: Union[np.ndarray, Image.Image, torch.Tensor, str]
+    img: Union[Image.Image, torch.Tensor, str]
   output:
     img_transformed: torch.Tensor
\ No newline at end of file