                                                              
from transformers import MarianMTModel, MarianTokenizer

model_name = "chi-vi/hirashiba-mt-tiny-zh-vi"
tokenizer = MarianTokenizer.from_pretrained(model_name)
model = MarianMTModel.from_pretrained(model_name)

def translate(lines):
    inputs = tokenizer(lines, return_tensors="pt", padding=True)
    translated = model.generate(**inputs)
    return [tokenizer.decode(t, skip_special_tokens=True) for t in translated]

with open('sample.txt') as f:
    src_text = f.readlines()

import time

start = time.time()
translated = translate(src_text)
end = time.time()

print(translated)
print(end - start)