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='josu/gpt-neo-pt-br')
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'}]
