Update app.py

app.py

@@ -3,7 +3,7 @@ from transformers import pipeline
 from typing import Union, List
 
 class BambaraTranslator:
-    def __init__(self, model_name: str = "sudoping01/nllb-bambara-v2"):
+    def __init__(self, model_name: str ="sudoping01/nllb-200-600M-bambara-nko" ): #"sudoping01/nllb-bambara-v2"):
         self.translator = pipeline(
             "translation",
             model=model_name,
@@ -14,7 +14,7 @@ class BambaraTranslator:
         self.flores_codes = {
             "French": "fra_Latn",
             "English": "eng_Latn",
-            "Bambara": "bam_Latn"
+            "Bambara": 'nqo_Nkoo' #"bam_Latn"
         }
 
     def translate(self, text: Union[str, List[str]], src_lang: str, tgt_lang: str) -> Union[str, List[str]]: