src/model_registry.py

MODEL_REGISTRY = {
    # multilingual (best for code-mix)
    "xlmr": "xlm-roberta-base",

    # Bangla
    "banglabert": "csebuetnlp/banglabert",

    # Indic multilingual
    "indicbert": "ai4bharat/indic-bert",

    # English BERT baseline
    "bert": "bert-base-uncased",

    # Mental-health domain (English)
    # NOTE: change this to the exact HuggingFace repo you prefer/are using
    "mentalbert": "mental/mental-bert-base-uncased",
}