Update goai_helpers/goai_traduction.py
Browse files
goai_helpers/goai_traduction.py
CHANGED
|
@@ -60,6 +60,7 @@ def goai_traduction(text, src_lang, tgt_lang):
|
|
| 60 |
no_repeat_ngram_size=0,
|
| 61 |
length_penalty=2.0
|
| 62 |
)
|
|
|
|
| 63 |
|
| 64 |
return translation
|
| 65 |
|
|
|
|
| 60 |
no_repeat_ngram_size=0,
|
| 61 |
length_penalty=2.0
|
| 62 |
)
|
| 63 |
+
translation = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
|
| 64 |
|
| 65 |
return translation
|
| 66 |
|