Process generator input

Signed-off-by: Kaiyuan Hu <kaiyuan.hu@zilliz.com>
2 years ago · f5aaced8d5
1 changed files with 2 additions and 1 deletions
--- a/isc.py
+++ b/isc.py
@ -69,10 +69,11 @@ class Isc(NNOperator):
        ])

    def __call__(self, data: Union[List[towhee._types.Image], towhee._types.Image]):
-        if not isinstance(data, list):
+        if isinstance(data, towhee._types.Image):
            imgs = [data]
        else:
            imgs = data
+
        img_list = []
        for img in imgs:
            img = self.convert_img(img)