Update README.md

README.md

@@ -27,7 +27,8 @@ The model was trained on wiki-corpus for 40 hours using NVIDIA Tesla P100 GPU.
 
 ### Load model
 
-``` from transformers import AutoTokenizer, TFAutoModelWithLMHead
+``` 
+from transformers import AutoTokenizer, TFAutoModelWithLMHead
 import tensorflow as tf
 
 tokenizer = AutoTokenizer.from_pretrained("DeividasM/gpt2_lithuanian_small")
@@ -40,7 +41,8 @@ model.eval()
 ```
 ## Generate text
 
-``` text = "tekstas"
+``` 
+text = "tekstas "
 inputs = tokenizer.encode(text, return_tensors="tf")