2021-08-08 21:43:24 +01:00
|
|
|
import os
|
2021-08-11 14:11:34 +01:00
|
|
|
import shutil
|
2021-08-08 21:43:24 +01:00
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
|
import PyInstaller.__main__
|
2021-08-11 14:11:34 +01:00
|
|
|
from PyInstaller.utils.hooks import get_package_paths
|
2021-08-08 21:43:24 +01:00
|
|
|
|
2021-08-11 14:11:34 +01:00
|
|
|
pytorch_libs = os.path.join(get_package_paths('torch')[1], "lib")
|
2021-08-08 21:43:24 +01:00
|
|
|
|
2021-08-11 14:11:34 +01:00
|
|
|
command = [
|
2021-08-08 21:43:24 +01:00
|
|
|
"main.py",
|
|
|
|
|
"--onefile",
|
|
|
|
|
"--clean",
|
2021-08-11 14:11:34 +01:00
|
|
|
"--icon=application\static\\favicon\\app-icon.ico",
|
|
|
|
|
"--add-data=application/static;application/static",
|
|
|
|
|
"--additional-hooks=extra-hooks",
|
2021-08-08 21:43:24 +01:00
|
|
|
"--hidden-import=sklearn.utils._weight_vector",
|
|
|
|
|
"--hidden-import=sklearn.utils._cython_blas",
|
|
|
|
|
"--hidden-import=sklearn.neighbors._typedefs",
|
|
|
|
|
"--hidden-import=sklearn.neighbors._quad_tree",
|
|
|
|
|
"--hidden-import=sklearn.tree._utils",
|
|
|
|
|
"--hidden-import=sklearn.tree",
|
|
|
|
|
"--hidden-import=scipy.special.cython_special",
|
|
|
|
|
"--add-data",
|
2021-08-11 14:11:34 +01:00
|
|
|
pytorch_libs+";."
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
PyInstaller.__main__.run(command)
|
|
|
|
|
shutil.rmtree("build")
|
