Update parameters

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
2 years ago · f76c2cf953
1 changed files with 2 additions and 2 deletions
--- a/image_decode.py
+++ b/image_decode.py
@ -7,8 +7,8 @@ from towhee.operator import Operator
@register(output_schema='image')
 class ImageDecode(Operator):

-    def __init__(self):
-        self._op = ops.image_decode.cv2()
+    def __init__(self, *args, **kwargs):
+        self._op = ops.image_decode.cv2(*args, **kwargs)

    def __call__(self, image_path: str):
        return self._op(image_path)