cancel import warning

3 years ago · 22aa3c7e26
1 changed files with 2 additions and 1 deletions
--- 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):
    """