Update goai_helpers/goai_traduction.py
Browse files
goai_helpers/goai_traduction.py
CHANGED
|
@@ -57,7 +57,7 @@ def translate_chunk(text, src_lang, tgt_lang):
|
|
| 57 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 58 |
|
| 59 |
if src_lang == "mos_Latn" and tgt_lang == "fra_Latn":
|
| 60 |
-
model_id = "ArissBandoss/
|
| 61 |
else:
|
| 62 |
model_id = "ArissBandoss/nllb-200-distilled-600M-finetuned-fr-to-mos-V4"
|
| 63 |
|
|
|
|
| 57 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 58 |
|
| 59 |
if src_lang == "mos_Latn" and tgt_lang == "fra_Latn":
|
| 60 |
+
model_id = "ArissBandoss/mos2fr-3B"
|
| 61 |
else:
|
| 62 |
model_id = "ArissBandoss/nllb-200-distilled-600M-finetuned-fr-to-mos-V4"
|
| 63 |
|