diff --git a/modelscope/models/cv/image_classification/mmcls_model.py b/modelscope/models/cv/image_classification/mmcls_model.py index 52c0568b..6a65656e 100644 --- a/modelscope/models/cv/image_classification/mmcls_model.py +++ b/modelscope/models/cv/image_classification/mmcls_model.py @@ -27,9 +27,9 @@ class ClassificationModel(TorchModel): self.load_pretrained_checkpoint() - def forward(self, Inputs): + def forward(self, inputs): - return self.cls_model(**Inputs) + return self.cls_model(**inputs) def load_pretrained_checkpoint(self): import mmcv