Update goai_helpers/goai_traduction.py

goai_helpers/goai_traduction.py

@@ -56,13 +56,6 @@ def goai_traduction(text, src_lang, tgt_lang, max_chunk_length=80):
 
 def translate_chunk(text, src_lang, tgt_lang):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-
-    number_converter = MooreConverter()
-    numbers = re.findall(r'\b\d+\b', text)
-    for number in numbers:
-        moore_number = number_converter.number_to_moore(int(number))
-        if moore_number:  # Only replace if conversion succeeded
-            text = text.replace(number,moore_number)
     
     if src_lang == "mos_Latn" and tgt_lang == "fra_Latn":
         model_id = "ArissBandoss/nllb-200-3.3B-mos2fr"
@@ -92,6 +85,13 @@ def translate_chunk(text, src_lang, tgt_lang):
     
     # Décodage
     translation = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
+
+    number_converter = MooreConverter()
+    numbers = re.findall(r'\b\d+\b', translation)
+    for number in numbers:
+        moore_number = number_converter.number_to_moore(int(number))
+        if moore_number :
+            translation = translation.replace(number,moore_number)
     
     return translation