Update app/config.py
Browse files- 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", "
|
| 10 |
)
|
| 11 |
|
| 12 |
# Language detection
|
| 13 |
FASTTEXT_MODEL: str = "facebook/fasttext-language-identification"
|
| 14 |
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
|
| 18 |
-
|
| 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 |
-
|
| 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 |
-
|
| 38 |
-
|
| 39 |
-
"yoruba": "
|
| 40 |
-
"igbo": "
|
| 41 |
-
"hausa": "
|
| 42 |
-
"english": "
|
| 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 |
|