Update main.py

main.py

@@ -25,7 +25,7 @@ print(f"Using device: {device}")
 # =====================
 # Charger le modèle et le tokenizer NLLB
 try:
-    model_name = "Gaoussin/bamalingua-2"
+    model_name = "Gaoussin/bamalingua-4"
     tokenizer = NllbTokenizer.from_pretrained(model_name)
     # Move model to the selected device (CPU or GPU)
     model = AutoModelForSeq2SeqLM.from_pretrained(model_name).to(device)