File size: 459 Bytes
e3fbfd2 |
1 2 3 4 5 6 7 8 9 10 11 |
# export KMP_DUPLICATE_LIB_OK=TRUE
import ctranslate2
import sentencepiece as spm
from translate import Translator
text="Chiamo dall'ufficio fiscale"
translator = ctranslate2.Translator("it_en/")
sp = spm.SentencePieceProcessor(model_file="it_en/sentencepiece.model")
tokens = sp.encode(text, out_type=str)
translated_tokens = translator.translate_batch([tokens])
output = sp.decode(translated_tokens[0][0]["tokens"]).replace("▁"," " )
print( f"{output}") |