I have the following structure of folders
project\\ main.py datasets\\ __init__.py custom_data.py models