From bbea494c5ea1e0ed69631220df55e1d55eddf810 Mon Sep 17 00:00:00 2001
From: gexy5 <xinyu.ge@zilliz.com>
Date: Fri, 17 Jun 2022 15:50:12 +0800
Subject: [PATCH] add uniformer

Signed-off-by: gexy5 <xinyu.ge@zilliz.com>
---
 uniformer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/uniformer.py b/uniformer.py
index 037c0da..0f7b517 100644
--- a/uniformer.py
+++ b/uniformer.py
@@ -16,7 +16,7 @@ from towhee.models.uniformer.uniformer import create_model
 log = logging.getLogger()
 
 @register(output_schema=['labels', 'scores', 'features'])
-class Tsm(NNOperator):
+class Uniformer(NNOperator):
     """
     Generate a list of class labels given a video input data.
     Default labels are from [Kinetics400 Dataset](https://deepmind.com/research/open-source/kinetics).