drrobot9 commited on
Commit
89eba48
·
verified ·
1 Parent(s): aa14af4

Update app/config.py

Browse files
Files changed (1) hide show
  1. app/config.py +12 -13
app/config.py CHANGED
@@ -6,22 +6,21 @@ from typing import Dict
6
  @dataclass
7
  class Config:
8
  HF_TOKEN: str = field(
9
- default_factory=lambda: os.environ.get("HF_TOKEN", "HF_TOKEN").strip()
10
  )
11
 
12
  # Language detection
13
  FASTTEXT_MODEL: str = "facebook/fasttext-language-identification"
14
 
15
- # Translation
16
- TRANSLATION_MODEL_TO_ENG: str = "facebook/nllb-200-3.3B"
17
- TRANSLATION_MODEL_FROM_ENG: str = "facebook/nllb-200-3.3B"
18
- BASE_TRANSLATION: str = 'drrobot9/nllb-ig-yo-ha-finetuned'
19
-
20
  LLM_MODEL: str = "Qwen/Qwen2.5-1.5B-Instruct"
21
  LLM_MAX_TOKENS: int = 512
22
  LLM_TEMPERATURE: float = 0.2
23
 
24
- # ASR one model per language, selected by user
25
  ASR_MODELS: Dict[str, str] = field(default_factory=lambda: {
26
  "yoruba": "NCAIR1/Yoruba-ASR",
27
  "igbo": "NCAIR1/Igbo-ASR",
@@ -34,12 +33,12 @@ class Config:
34
  # Memory
35
  MAX_HISTORY_TURNS: int = 10
36
 
37
- # language codes for models
38
- LANG_CODES: Dict[str, str] = field(default_factory=lambda: {
39
- "yoruba": "yo",
40
- "igbo": "ig",
41
- "hausa": "ha",
42
- "english": "en",
43
  })
44
 
45
 
 
6
  @dataclass
7
  class Config:
8
  HF_TOKEN: str = field(
9
+ default_factory=lambda: os.environ.get("HF_TOKEN", "").strip()
10
  )
11
 
12
  # Language detection
13
  FASTTEXT_MODEL: str = "facebook/fasttext-language-identification"
14
 
15
+
16
+ TRANSLATION_MODEL: str = "facebook/nllb-200-3.3B"
17
+
18
+ # LLM
 
19
  LLM_MODEL: str = "Qwen/Qwen2.5-1.5B-Instruct"
20
  LLM_MAX_TOKENS: int = 512
21
  LLM_TEMPERATURE: float = 0.2
22
 
23
+
24
  ASR_MODELS: Dict[str, str] = field(default_factory=lambda: {
25
  "yoruba": "NCAIR1/Yoruba-ASR",
26
  "igbo": "NCAIR1/Igbo-ASR",
 
33
  # Memory
34
  MAX_HISTORY_TURNS: int = 10
35
 
36
+
37
+ NLLB_CODES: Dict[str, str] = field(default_factory=lambda: {
38
+ "yoruba": "yor_Latn",
39
+ "igbo": "ibo_Latn",
40
+ "hausa": "hau_Latn",
41
+ "english": "eng_Latn",
42
  })
43
 
44