dl-hack-distilgpt2 / README.md
vasudevgupta's picture
Update README.md
1c19c37

DL research papers Title -> abstract

Using this model

from transformers import pipeline, GPT2LMHeadModel, GPT2Tokenizer

tokenizer = GPT2Tokenizer.from_pretrained("vasudevgupta/dl-hack-distilgpt2")
model = GPT2LMHeadModel.from_pretrained("vasudevgupta/dl-hack-distilgpt2")
agent = pipeline("text-generation", model=model, tokenizer=tokenizer)

print(agent("An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale", max_length=200))