diff --git a/image_decode.py b/image_decode.py index fd31fbc..1805d3e 100644 --- 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)