transZ commited on
Commit
9c2c4d0
1 Parent(s): f20020d

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -0
README.md ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # How to run the model
2
+ ```python
3
+ from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
4
+
5
+ model = M2M100ForConditionalGeneration.from_pretrained("transZ/M2M_Vi_Ba")
6
+ tokenizer = M2M100Tokenizer.from_pretrained("transZ/M2M_Vi_Ba")
7
+ tokenizer.src_lang = "vi"
8
+ vi_text = "Hôm nay ba đi chợ."
9
+ encoded_vi = tokenizer(vi_text, return_tensors="pt")
10
+ generated_tokens = model.generate(**encoded_vi, forced_bos_token_id=tokenizer.get_lang_id("ba"))
11
+ translate = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
12
+ print(translate)
13
+ ```