akera commited on
Commit
11a64ab
·
verified ·
1 Parent(s): fae38f8

Create config.py

Browse files
Files changed (1) hide show
  1. config.py +42 -0
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
+ }