Update README.md

README.md

@@ -147,6 +147,8 @@ Training data can be provided upon request.
 
 - **Generation**
 
+Beam size of 5, and maximum target length of 256 is used for the generation.
+
 ```
 from transformers import M2M100ForConditionalGeneration
 from tokenization_small100 import SMALL100Tokenizer
@@ -160,14 +162,14 @@ tokenizer = SMALL100Tokenizer.from_pretrained("alirezamsh/small100")
 # translate Hindi to French
 tokenizer.tgt_lang = "fr"
 encoded_hi = tokenizer(hi_text, return_tensors="pt")
-generated_tokens = model.generate(**encoded_hi)
+generated_tokens = model.generate(**encoded_hi, max_length=256, num_beams=5)
 tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
 # => "La vie est comme une boîte de chocolat."
 
 # translate Chinese to English
 tokenizer.tgt_lang = "en"
 encoded_zh = tokenizer(chinese_text, return_tensors="pt")
-generated_tokens = model.generate(**encoded_zh)
+generated_tokens = model.generate(**encoded_zh, max_length=256, num_beams=5)
 tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
 # => "Life is like a box of chocolate."
 ```