diff --git a/ppocr/modeling/architectures/cls_model.py b/ppocr/modeling/architectures/cls_model.py
index 6df20770..ad3ad0e7 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