From 63467c6fdd3d1d2d82f3114afd04383d1a29f521 Mon Sep 17 00:00:00 2001
From: Jael Gu <mengjia.gu@zilliz.com>
Date: Thu, 31 Mar 2022 19:39:47 +0800
Subject: [PATCH] Update

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
---
 README.md   | 2 ++
 __init__.py | 4 ++--
 2 files changed, 4 insertions(+), 2 deletions(-)

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)