diff --git a/deploy/lite/fuse_gallery.py b/ppcls/utils/gallery2fc.py
similarity index 95%
rename from deploy/lite/fuse_gallery.py
rename to ppcls/utils/gallery2fc.py
index cf00956b0cb67b9ef5ad083b564ecf7a96a7380b..8d9a502d623580cf18e5f40b5bab587ef2824784 100644
--- a/deploy/lite/fuse_gallery.py
+++ b/ppcls/utils/gallery2fc.py
@@ -1,4 +1,5 @@
 import paddle
+from ppcls.arch import build_model
 from deploy.utils.config import parse_config, parse_args