Update README.md

README.md

@@ -8,9 +8,17 @@ widget:
 ---
 
 ``` python
+
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+model = AutoModelForCausalLM.from_pretrained("josu/gpt-neo-pt-br")
+tokenizer = AutoTokenizer.from_pretrained("josu/gpt-neo-pt-br")
+
 from transformers import pipeline
 
 generator = pipeline('text-generation', model='gpt2')
+
 generator("Futebol é o esporte,", max_length=30, num_return_sequences=1)
+
 [{'generated_text': 'Futebol é o esporte, sendo que o país é um dos esportes mais populares do mundo. A seleção nacional de futebol do país é o mais popular e'}]
   ```