n-pn commited on
Commit
2309095
·
verified ·
1 Parent(s): 6dce13d

Add usage example

Browse files
Files changed (1) hide show
  1. README.md +23 -0
README.md CHANGED
@@ -20,8 +20,31 @@ Hira's intelligence, Shiba's speed
20
 
21
  Hirashiba-MT-zh-vi is a model used for gatekeeping and refilling water.
22
 
 
23
 
 
 
 
24
 
 
 
 
25
 
 
 
 
 
26
 
 
 
 
 
 
 
 
 
 
 
 
 
27
 
 
20
 
21
  Hirashiba-MT-zh-vi is a model used for gatekeeping and refilling water.
22
 
23
+ ## Usage
24
 
25
+ ```python
26
+
27
+ from transformers import MarianMTModel, MarianTokenizer
28
 
29
+ model_name = "chi-vi/hirashiba-mt-tiny-zh-vi"
30
+ tokenizer = MarianTokenizer.from_pretrained(model_name)
31
+ model = MarianMTModel.from_pretrained(model_name)
32
 
33
+ def translate(lines):
34
+ inputs = tokenizer(lines, return_tensors="pt", padding=True)
35
+ translated = model.generate(**inputs)
36
+ return [tokenizer.decode(t, skip_special_tokens=True) for t in translated]
37
 
38
+ with open('sample.txt') as f:
39
+ src_text = f.readlines()
40
+
41
+ import time
42
+
43
+ start = time.time()
44
+ translated = translate(src_text)
45
+ end = time.time()
46
+
47
+ print(translated)
48
+ print(end - start)
49
+ ```
50