From 52509e900a7d4192ce990526a8da2d3863d928ac Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Mon, 28 Mar 2022 14:36:49 +0800 Subject: [PATCH] Update README Signed-off-by: Jael Gu --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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) ```