diff --git a/README.md b/README.md index 833dc2c..ca1a252 100644 --- a/README.md +++ b/README.md @@ -14,7 +14,7 @@ An image embedding operator implemented with pretrained models provided by [Timm from towhee import ops import numpy as np -img_encoder = ops.image_embedding.timm('resnet50') +img_encoder = ops.image_embedding.timm(model_name='resnet50') fake_img = np.zeros((256, 256, 3)) image_embedding = img_encoder(fake_img) ```