Spaces:
Runtime error
Runtime error
File size: 496 Bytes
6796f63 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from transformers import pipeline, GPT2Tokenizer
# Load the pre-trained GPT-2 model and tokenizer
model = pipeline("text-generation", model="gpt2")
tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
# Generate an image from input text
text = "A blue bird flying in the sky"
input_ids = tokenizer.encode(text, return_tensors="pt")
output = model.generate(input_ids=input_ids, max_length=1024, do_sample=True)
image = generate_image_from_output(output)
# Display the generated image
image.show()
|