Update config.py
Browse files
config.py
CHANGED
|
@@ -1,11 +1,22 @@
|
|
| 1 |
# Model configuration
|
| 2 |
-
|
| 3 |
-
|
| 4 |
-
|
| 5 |
-
|
| 6 |
-
"
|
| 7 |
-
|
| 8 |
-
|
| 9 |
-
|
| 10 |
-
}
|
| 11 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
# Model configuration
|
| 2 |
+
MODEL_NAME = "facebook/m2m100_1.2B"
|
| 3 |
+
|
| 4 |
+
# Language mapping with UNESCO status
|
| 5 |
+
LANGUAGE_MAPPING = {
|
| 6 |
+
"English": {"code": "en", "status": "Safe"},
|
| 7 |
+
"Spanish": {"code": "es", "status": "Safe"},
|
| 8 |
+
"French": {"code": "fr", "status": "Safe"},
|
| 9 |
+
"Quechua": {"code": "qu", "status": "Vulnerable"},
|
| 10 |
+
"Aymara": {"code": "ay", "status": "Vulnerable"},
|
| 11 |
+
"Cherokee": {"code": "chr", "status": "Endangered"},
|
| 12 |
+
"Navajo": {"code": "nv", "status": "Vulnerable"},
|
| 13 |
+
"Inuktitut": {"code": "iu", "status": "Vulnerable"},
|
| 14 |
+
"Sami": {"code": "se", "status": "Endangered"},
|
| 15 |
+
"Welsh": {"code": "cy", "status": "Vulnerable"}
|
| 16 |
+
}
|
| 17 |
+
|
| 18 |
+
# Endangered languages for target selection
|
| 19 |
+
ENDANGERED_LANGS = [
|
| 20 |
+
"Quechua", "Aymara", "Cherokee",
|
| 21 |
+
"Navajo", "Inuktitut", "Sami", "Welsh"
|
| 22 |
+
]
|