Update app.py

app.py

@@ -18,8 +18,8 @@ from tts import BambaraTTS
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # Translation pipeline 
-# translation_model = "oza75/nllb-600M-mt-french-bambara"
-translation_model = "oza75/nllb-1.3B-mt-french-bambara"
+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