File size: 381 Bytes
545f8ff
faca2af
6db0654
b57ee92
07affc8
b57ee92
 
07affc8
b57ee92
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16



from transformers import pipeline, set_seed

# Load the GPT-2 medium model
generator = pipeline('text-generation', model='gpt2-medium')

# Set a seed for reproducibility (optional)
set_seed(42)

# Generate text
prompt = "GPT-2 is a model developed by OpenAI."
generated_text = generator(prompt, max_length=100, num_return_sequences=1)[0]['generated_text']

print(generated_text)