Update app.py

app.py

@@ -21,8 +21,8 @@ else:
     logger.info("Using CPU for inference.")
 
 # Load the translation pipeline with the specified model and detected device
-# model_checkpoint = "oza75/bm-nllb-1.3B"
-model_checkpoint = "oza75/nllb-600M-mt-french-bambara"
+model_checkpoint = "oza75/bm-nllb-1.3B"
+# model_checkpoint = "oza75/nllb-600M-mt-french-bambara"
 translator = pipeline("translation", model=model_checkpoint, device=device, max_length=512)
 logger.info("Translation pipeline initialized successfully.")