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