diff --git a/src/const.py b/src/const.py index 57c10e7..148f92b 100644 --- a/src/const.py +++ b/src/const.py @@ -8,7 +8,7 @@ from image import Image from machine_learning.neural_network.net import Net -main_path = os.path.dirname(os.getcwd()) +main_path = os.path.dirname(os.path.dirname(__file__)) WIDTH = 800 HEIGHT = 800