Spaces:
Runtime error
Runtime error
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) |