| # Import necessary libraries | |
| from transformers import DALL_E | |
| from PIL import Image | |
| import requests | |
| # Load the pre-trained DALL-E model | |
| model = DALL_E.from_pretrained("openai/your-dall-e-model-name-here") | |
| # Define a text description for the image you want to generate | |
| text_description = "A cat sitting on a grassy hill under a blue sky" | |
| # Generate an image from the text description | |
| output = model.generate_images(text_description) | |
| # Get the image data | |
| image_data = requests.get(output[0]["image"]).content | |
| # Save the image to a file or display it | |
| with open("generated_image.jpg", "wb") as img_file: | |
| img_file.write(image_data) | |
| # Display the image using PIL (optional) | |
| image = Image.open("generated_image.jpg") | |
| image.show() |