Views
No views yet
1from transformers import GPT2LMHeadModel, GPT2Tokenizer
2
3model = GPT2LMHeadModel.from_pretrained("pkailin2002/gpt2-tuned")
4tokenizer = GPT2Tokenizer.from_pretrained("pkailin2002/gpt2-tuned")
5
6# Generate text
7input_text = "Your prompt here"
8inputs = tokenizer.encode(input_text, return_tensors="pt")
9outputs = model.generate(inputs, max_length=100, num_return_sequences=1, temperature=0.7)
10generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
11print(generated_text)