diff --git a/image_embedding_resnet50.py b/image_embedding_resnet50.py
index 27de7a7..a863783 100644
--- 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())