From stable_diffusion import StableDiffusionPipeline import torch from PIL impor

from stable_diffusion import StableDiffusionPipeline import torch from PIL import Image # Load the Stable Diffusion model pipe = StableDiffusionPipeline.from_pretrained("stable-diffusion-v1-4") pipe = pipe.to("cuda") # Use GPU if available # Define the prompt for Magical Girls, Adventure, Romance, and Shoujo themes with Bitcoin, Whale, and Shark prompts = { "magical_girls_adventure_romance_bitcoin_whale_shark": ( "In a whimsical, enchanting world where magic and adventure intertwine, a group of magical girls with sparkling outfits and wands embark on a quest filled with excitement and romance. " "One of the magical girls, wearing a costume adorned with Bitcoin symbols, stands on the deck of a magical ship, gazing at a majestic, benevolent whale that glows with a golden Bitcoin chain around its neck. " "The whale, with a friendly and wise expression, offers guidance and support to the girls. " "Nearby, a sleek, futuristic shark with Bitcoin patterns on its fin swims through the crystalline waters, adding a touch of danger and thrill. " "The scene is filled with vibrant colors, magical light, and a sense of wonder, capturing the adventurous spirit and romantic undertones of the story. " "The background features fantastical landscapes and sparkling magical effects, emphasizing the dreamy and romantic elements of the narrative." ) } # Generate the image for the prompt for name, prompt in prompts.items(): # Generate the image using the prompt image = pipe(prompt).images[0] # Save the generated image filename = f"{name.lower().replace(' ', '_')}.png" image.save(filename) print(f"Saved image: {filename}") # Optionally, show the generated image image.show()