├── README.md
└── main.py


/README.md:
--------------------------------------------------------------------------------
1 | # Nlp-Project


--------------------------------------------------------------------------------
/main.py:
--------------------------------------------------------------------------------
 1 | import torch
 2 | from transformers import AutoTokenizer, GPT2DoubleHeadsModel
 3 | 
 4 | tokenizer = AutoTokenizer.from_pretrained("gpt2")
 5 | model = GPT2DoubleHeadsModel.from_pretrained("gpt2")
 6 | 
 7 | num_added_tokens = tokenizer.add_special_tokens({"cls_token": "[CLS]"})
 8 | 
 9 | prompt = input("Enter your prompt: ")
10 | 
11 | input_ids = tokenizer.encode(prompt, return_tensors="pt")
12 | 
13 | output = model.generate(input_ids)
14 | 
15 | output_text = tokenizer.decode(output[0], skip_special_tokens=True)
16 | 
17 | print(output_text)
18 | 


--------------------------------------------------------------------------------