models/model_v3/config.py

import os

BASE_DIR = os.path.dirname(os.path.abspath(__file__))

# Hugging Face Configuration
HF_REPO_ID = "cracker0935/adtrackv3"
WEIGHTS_FILENAME = "multimodal_dementia_model.pth"

# Local fallback path
LOCAL_WEIGHTS_PATH = os.path.join(BASE_DIR, WEIGHTS_FILENAME)

# Model Configuration
TEXT_MODEL_NAME = "microsoft/deberta-base"
MAX_LEN = 128
WHISPER_MODEL_SIZE = "base"

# Segmentation Dataset Path (for auto-discovery of segmentation CSVs)
# Structure: SEGMENTATION_ROOT_PATH/AD/filename.csv and SEGMENTATION_ROOT_PATH/Control/filename.csv
# Set to None to disable auto-discovery (user must upload segmentation file)
SEGMENTATION_ROOT_PATH = os.environ.get("SEGMENTATION_ROOT_PATH", "models/model_v3/segmentation")