File size: 318 Bytes
a3a57c6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
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"] |