diff --git a/README.md b/README.md index 6283098..8e2c701 100644 --- a/README.md +++ b/README.md @@ -38,6 +38,8 @@ dc.glob['path']('./dog.jpg') \ .select('vec') \ .to_list() ``` + [array([0. , 0. , 0. , ..., 0. , 0.01748613, + 0. ], dtype=float32)] ## Factory Constructor diff --git a/__init__.py b/__init__.py index c5a3c9c..a25960d 100644 --- a/__init__.py +++ b/__init__.py @@ -15,5 +15,5 @@ from .timm_image import TimmImage -def timm(model_name: str, num_classes: int = 1000, skip_preprocess: bool = False): - return TimmImage(model_name=model_name, num_classes=num_classes, skip_preprocess=skip_preprocess) +def timm(**kwargs): + return TimmImage(kwargs)