diff --git a/efficientnet_image_embedding.py b/efficientnet_image_embedding.py index 1af7a5a..11e7661 100644 --- a/efficientnet_image_embedding.py +++ b/efficientnet_image_embedding.py @@ -24,6 +24,8 @@ from towhee.operator import Operator from timm.data import resolve_data_config from timm.data.transforms_factory import create_transform import os +import warnings +warnings.filterwarnings("ignore") class EfficientnetImageEmbedding(Operator): """