diff --git a/README.md b/README.md index 7eba39a..a58d0d3 100644 --- a/README.md +++ b/README.md @@ -64,7 +64,7 @@ model_name='omnivore_swinT', skip_preprocess=False, classmap=None, topk=5)*** ​ ***model_name***: *str* -​ The name of pre-trained tsm model. +​ The name of pre-trained uniformer model. ​ Supported model names: - omnivore_swinT diff --git a/__init__.py b/__init__.py index 0dbfa53..5d36630 100644 --- a/__init__.py +++ b/__init__.py @@ -16,4 +16,4 @@ from .omnivore import Omnivore def omnivore(**kwargs): - return Omnivore(**kwargs) \ No newline at end of file + return Omnivore(**kwargs)