Update py

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
2 years ago · ee6f4c286d
1 changed files with 0 additions and 6 deletions
--- a/image_embedding_resnet50.py
+++ b/image_embedding_resnet50.py
@ -2,17 +2,11 @@ from towhee import pipe, ops, AutoPipes, AutoConfig

@AutoConfig.register
 class ImageEmbeddingConfig:
-    """
-    For UT
-    """
    def __init__(self):
        self.model = 'resnet50'

@AutoPipes.register
 def ImageEmbedding(config):
-    """
-    For UT
-    """
    return (
        pipe.input('path')
        .map('path', 'img', ops.image_decode())