demo.py references src.utils.load_model, but the file is located in the root directory as load_model.py.