Test1Tuwaiq / app.py
Faisal-Data's picture
Update app.py
faca2af verified
raw
history blame contribute delete
381 Bytes
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)