Moore-Language-Space-ZeroGPU

Sleeping

ArissBandoss commited on May 19

Commit

0d45087

verified ·

1 Parent(s): 437a86e

Update goai_helpers/goai_traduction.py

Files changed (1) hide show

goai_helpers/goai_traduction.py CHANGED Viewed

@@ -57,7 +57,7 @@ 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/3b-new-400"
     else:
         model_id = "ArissBandoss/nllb-200-distilled-600M-finetuned-fr-to-mos-V4"

     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     if src_lang == "mos_Latn" and tgt_lang == "fra_Latn":
+        model_id = "ArissBandoss/3b-neew-600"
     else:
         model_id = "ArissBandoss/nllb-200-distilled-600M-finetuned-fr-to-mos-V4"