Update app.py

app.py

@@ -17,8 +17,9 @@ from tts import BambaraTTS
 # Check if CUDA is available
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-# Translation pipeline
-translation_model = "oza75/nllb-600M-mt-french-bambara"
+# Translation pipeline 
+# translation_model = "oza75/nllb-600M-mt-french-bambara"
+translation_model = "oza75/nllb-1.3B-mt-french-bambara"
 translator = pipeline("translation", model=translation_model, max_length=512)
 
 # Text-to-Speech pipeline