Update app.py

app.py

@@ -38,6 +38,8 @@ class Language(str, Enum):
     WOLOF = "Wolof"
     HAUSA = "Hausa"
     DARIJA = "Darija"
+    SWAHILI = "Swahili"
+    BAMBARA = "Bambara"
 
 class InputMode(str, Enum):
     """Supported input modes."""
@@ -67,7 +69,19 @@ TRANSLATION_MODELS: Dict[Tuple[Language, Language], ModelConfig] = {
     ),
     (Language.ENGLISH, Language.DARIJA): ModelConfig(
         "LocaleNLP/english_darija", ">>dar<<"
-    )
+    ),
+    (Language.ENGLISH, Language.BAMBARA): ModelConfig(
+        "LocaleNLP/localenlp-eng-bam-0.03", ">>bam<<"
+    ),
+    (Language.BAMBARA, Language.ENGLISH): ModelConfig(
+        "LocaleNLP/localenlp-bam-eng-0.03", ">>eng<<"
+    ),
+    (Language.SWAHILI, Language.ENGLISH): ModelConfig(
+        "LocaleNLP/localenlp-swa-eng-0.03", ">>eng<<"
+    ),
+    (Language.ENGLISH, Language.SWAHILI): ModelConfig(
+        "LocaleNLP/localenlp-eng-swa-0.03", ">>swa<<"
+    ),
 }
 
 # File type support
@@ -132,7 +146,7 @@ class ModelManager:
             logger.info(f"Loading translation model: {config.model_name}")
             
             # Authenticate with Hugging Face if token provided
-            if hf_token := os.getenv("hffff"):
+            if hf_token := os.getenv("final_tk"):
                 login(token=hf_token)
             
             model = AutoModelForSeq2SeqLM.from_pretrained(
@@ -476,7 +490,7 @@ class TranslationApp:
             # Header
             gr.Markdown("""
             # 🌍 LocaleNLP Translation Service
-            Translate between English, Wolof, Hausa, and Darija with support for text, audio, and documents.
+            Translate between English, Wolof, Hausa,Bambara, Swahili and Darija with support for text, audio, and documents.
             """)
             
             # Input controls