From 838738f354380579cf2ae324611abb1459efcc00 Mon Sep 17 00:00:00 2001 From: shiyu22 Date: Wed, 15 Dec 2021 21:15:58 +0800 Subject: [PATCH] Update Signed-off-by: shiyu22 --- resnet_image_embedding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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. """