Spaces:
Sleeping
Sleeping
Create config.py
Browse files
config.py
ADDED
@@ -0,0 +1,42 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# config.py
|
2 |
+
import os
|
3 |
+
|
4 |
+
# HuggingFace settings
|
5 |
+
HF_TOKEN = os.getenv("HF_TOKEN") # Set in Space secrets
|
6 |
+
LEADERBOARD_DATASET = "Sunbird/salt-translation-leaderboard"
|
7 |
+
SALT_DATASET = "sunbird/salt"
|
8 |
+
|
9 |
+
# Model settings
|
10 |
+
MAX_MODEL_SIZE_GB = 15 # Limit for HF Space
|
11 |
+
SUPPORTED_MODEL_TYPES = [
|
12 |
+
"gemma", "qwen", "llama", "nllb", "google-translate"
|
13 |
+
]
|
14 |
+
|
15 |
+
# Evaluation settings
|
16 |
+
MAX_EVAL_SAMPLES = 200 # Limit for faster evaluation
|
17 |
+
BATCH_SIZE = 4
|
18 |
+
MAX_NEW_TOKENS = 100
|
19 |
+
|
20 |
+
# UI settings
|
21 |
+
TITLE = "🏆 SALT Translation Model Leaderboard"
|
22 |
+
DESCRIPTION = """
|
23 |
+
Evaluate your translation models on Ugandan languages!
|
24 |
+
Submit a HuggingFace model and see how it performs on Luganda, Acholi, and Swahili translation tasks.
|
25 |
+
"""
|
26 |
+
|
27 |
+
# Supported languages (Google Translate compatible subset)
|
28 |
+
SUPPORTED_LANGUAGES = ['lug', 'ach', 'swa', 'eng']
|
29 |
+
LANGUAGE_NAMES = {
|
30 |
+
'lug': 'Luganda',
|
31 |
+
'ach': 'Acholi',
|
32 |
+
'swa': 'Swahili',
|
33 |
+
'eng': 'English'
|
34 |
+
}
|
35 |
+
|
36 |
+
# Google Translate language mapping
|
37 |
+
GOOGLE_LANG_MAP = {
|
38 |
+
'lug': 'lg',
|
39 |
+
'ach': 'ach',
|
40 |
+
'swa': 'sw',
|
41 |
+
'eng': 'en'
|
42 |
+
}
|