diff --git a/resnet_image_embedding.py b/resnet_image_embedding.py
index a3401fd..d10f224 100644
--- a/resnet_image_embedding.py
+++ b/resnet_image_embedding.py
@@ -22,7 +22,7 @@ from typing import NamedTuple
 from towhee.operator import Operator
 
 
-class Resnet50ImageEmbedding(Operator):
+class ResnetImageEmbedding(Operator):
     """
     PyTorch model for image embedding.
     """