Update goai_helpers/goai_traduction.py

goai_helpers/goai_traduction.py

@@ -57,10 +57,10 @@ def translate_chunk(text, src_lang, tgt_lang):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     
     if src_lang == "mos_Latn" and tgt_lang == "fra_Latn":
-        model_id = "ArissBandoss/mos2fr-3B"
+        model_id = "ArissBandoss/nllb-200-3.3B-mos2fr"
     else:
-        #model_id = "ArissBandoss/nllb-200-distilled-600M-finetuned-fr-to-mos-V4"
-        model_id = "ArissBandoss/nllb-200-3.3B-mos-fr-bidirectional-peft"
+        model_id = "ArissBandoss/nllb-200-3.3B-fr2mos"
+        #model_id = "ArissBandoss/nllb-200-3.3B-mos-fr-bidirectional-peft"
     
     tokenizer = AutoTokenizer.from_pretrained(model_id, token=auth_token)
     model = AutoModelForSeq2SeqLM.from_pretrained(model_id, token=auth_token).to(device)