Update goai_helpers/goai_traduction.py

goai_helpers/goai_traduction.py

@@ -32,6 +32,7 @@ def goai_traduction(text, src_lang, tgt_lang):
     
     # Tokenisation
     inputs = tokenizer(text, return_tensors="pt", truncation=False).to(device)
+    print("Nombre de tokens :", inputs["input_ids"].shape[1])
     
     # ID du token de langue cible
     tgt_lang_id = tokenizer.convert_tokens_to_ids(tgt_lang)