diff --git a/timm_image.py b/timm_image.py index 09361e7..fe2cc97 100644 --- a/timm_image.py +++ b/timm_image.py @@ -30,10 +30,9 @@ from timm.models.factory import create_model import warnings warnings.filterwarnings('ignore') - - log = logging.getLogger() + @register(output_schema=['vec']) class TimmImage(NNOperator): """