diff --git a/resnet_image_embedding.py b/resnet_image_embedding.py index ba18e2d..98cb1a0 100644 --- a/resnet_image_embedding.py +++ b/resnet_image_embedding.py @@ -22,7 +22,8 @@ from typing import NamedTuple import os from torchvision.transforms import InterpolationMode from towhee.operator import Operator - +import warnings +warnings.filterwarnings("ignore") class ResnetImageEmbedding(Operator): """