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"]