diff --git a/python/paddle/hapi/model.py b/python/paddle/hapi/model.py
index 3bcc62e423bd5d5247cdccda0f57b8f7bc726c96..82ae908e2b10cc7dc642f1a61cf9fbc11307b521 100644
--- a/python/paddle/hapi/model.py
+++ b/python/paddle/hapi/model.py
@@ -969,6 +969,7 @@ class Model(object):
               import numpy as np
               import paddle
               import paddle.nn as nn
+              from paddle.static import InputSpec
 
               device = paddle.set_device('cpu') # or 'gpu'
               paddle.disable_static(device)
@@ -1093,6 +1094,7 @@ class Model(object):
             
               import paddle
               import paddle.nn as nn
+              from paddle.static import InputSpec
               
               device = paddle.set_device('cpu')
               paddle.disable_static(device)
@@ -1170,6 +1172,7 @@ class Model(object):
 
               import paddle
               import paddle.nn as nn
+              from paddle.static import InputSpec
 
               paddle.disable_static()