Navya-Sree's picture
Update config.py
3b7e3cf verified
raw
history blame contribute delete
795 Bytes
# Model configuration
MODEL_NAME = "facebook/m2m100_1.2B"
# Language mapping with UNESCO status
LANGUAGE_MAPPING = {
"English": {"code": "en", "status": "Safe"},
"Spanish": {"code": "es", "status": "Safe"},
"French": {"code": "fr", "status": "Safe"},
"Quechua": {"code": "qu", "status": "Vulnerable"},
"Aymara": {"code": "ay", "status": "Vulnerable"},
"Cherokee": {"code": "chr", "status": "Endangered"},
"Navajo": {"code": "nv", "status": "Vulnerable"},
"Inuktitut": {"code": "iu", "status": "Vulnerable"},
"Sami": {"code": "se", "status": "Endangered"},
"Welsh": {"code": "cy", "status": "Vulnerable"}
}
# Endangered languages for target selection
ENDANGERED_LANGS = [
"Quechua", "Aymara", "Cherokee",
"Navajo", "Inuktitut", "Sami", "Welsh"
]