Update goai_helpers/goai_traduction.py
Browse files
goai_helpers/goai_traduction.py
CHANGED
|
@@ -29,7 +29,7 @@ def goai_traduction(text, src_lang, tgt_lang):
|
|
| 29 |
tokenizer.src_lang = src_lang
|
| 30 |
|
| 31 |
# Tokenisation
|
| 32 |
-
inputs = tokenizer(text, return_tensors="pt").to(device)
|
| 33 |
|
| 34 |
# ID du token de langue cible
|
| 35 |
tgt_lang_id = tokenizer.convert_tokens_to_ids(tgt_lang)
|
|
|
|
| 29 |
tokenizer.src_lang = src_lang
|
| 30 |
|
| 31 |
# Tokenisation
|
| 32 |
+
inputs = tokenizer(text, return_tensors="pt", truncation=False).to(device)
|
| 33 |
|
| 34 |
# ID du token de langue cible
|
| 35 |
tgt_lang_id = tokenizer.convert_tokens_to_ids(tgt_lang)
|