dl-hack-gpt2-large / README.md
vasudevgupta's picture
Update README.md
0e1bafc

DL research papers Title -> abstract

Using this model

from transformers import pipeline, GPT2LMHeadModel, GPT2Tokenizer

tokenizer = GPT2Tokenizer.from_pretrained("vasudevgupta/dl-hack-gpt2-large")
model = GPT2LMHeadModel.from_pretrained("vasudevgupta/dl-hack-gpt2-large")
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))