diff --git a/paddlex/cv/models/explanation/visualize.py b/paddlex/cv/models/explanation/visualize.py
index 97505cc6a24e7f97662674141fffc7d156c06d6f..9da7ba381c9276c595f092d671ffdf97edae6f20 100644
--- a/paddlex/cv/models/explanation/visualize.py
+++ b/paddlex/cv/models/explanation/visualize.py
@@ -28,8 +28,7 @@ def visualize(img_file,
               num_samples=3000, 
               batch_size=50,
               save_dir='./'):
-    model_info = moedl.get_model_info()
-    if model_info['status'] != 'Normal':
+    if model.status != 'Normal':
         raise Exception('The explanation only can deal with the Normal model')
     model.arrange_transforms(
                 transforms=model.test_transforms, mode='test')