diff --git a/ppocr/modeling/architectures/cls_model.py b/ppocr/modeling/architectures/cls_model.py
index 6df20770bce189a2ccb60147e4fec8b3c89c861a..ad3ad0e7cf4010a14c70a700ed02d02ee1f1323b 100755
--- a/ppocr/modeling/architectures/cls_model.py
+++ b/ppocr/modeling/architectures/cls_model.py
@@ -79,6 +79,7 @@ class ClsModel(object):
             outputs = {'total_loss': loss, 'decoded_out': \
                 predicts['decoded_out'], 'label': label, 'acc': acc}
             return loader, outputs
-
+        elif mode == "export":
+            return [image, predicts]
         else:
             return loader, predicts