| import os | |
| from dotenv import load_dotenv | |
| load_dotenv() | |
| # Root dir | |
| ROOT_DIR = os.getcwd() | |
| # Model checkpoints and repo names | |
| MODEL_FILENAME = os.getenv("MODEL_FILENAME") | |
| HF_MODEL_REPO_NAME = os.getenv("HF_MODEL_REPO_NAME") | |
| # Other constants | |
| CLASSES = ["Alzheimer's Desease", "Mild Cognitive Impairment", "Control"] |