Update README.md

README.md

@@ -30,13 +30,17 @@ model_output=model.generate(inp, use_cache=True, num_beams=4, max_length=20, min
 # Decode to get output strings
 
 decoded_output=tokenizer.decode(model_output[0], skip_special_tokens=True, clean_up_tokenization_spaces=False)
+
 print(decoded_output) # I am a boy
 
 # What if we mask?
 
 inp = tokenizer("I am [MASK] </s> <2en>", add_special_tokens=False, return_tensors="pt", padding=True).input_ids
+
 model_output=model.generate(inp, use_cache=True, num_beams=4, max_length=20, min_length=1, early_stopping=True, pad_token_id=tokenizer.pad_token_id, decoder_start_token_id=tokenizer(["<2en>"], add_special_tokens=False).input_ids[0][0])
+
 decoded_output=tokenizer.decode(model_output[0], skip_special_tokens=True, clean_up_tokenization_spaces=False)
+
 print(decoded_output) # I am happy