metadata
library_name: transformers
license: gpl-3.0
datasets:
- Moleys/mtl-zh2vi
- Moleys/mtl-zh2vi-b
language:
- zh
- vi
metrics:
- bleu
pipeline_tag: translation
Hirashiba ^^
Hira's intelligence, Shiba's speed
Hirashiba-MT-zh-vi is a model used for gatekeeping and refilling water.
Usage
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)