Update main.py

main.py

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