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 Fantasy, Drama, and Psychological themes with Bitcoin, Whale, and Shark prompts = { "fantasy_drama_psychology_bitcoin_whale_shark": ( "In a fantastical, twilight realm where reality and dreams intertwine, a massive, majestic whale adorned with glowing Bitcoin symbols floats through a surreal sky. " "The whale exudes a sense of profound wisdom and calm, with its scales reflecting ethereal light and Bitcoin patterns. " "On a jagged, high cliff overlooking a stormy sea, a solitary figure in elaborate, flowing robes stands in deep contemplation. " "The figure's face is partially hidden in shadows, emphasizing the internal conflict and psychological depth. " "Nearby, a shadowy, menacing shark with Bitcoin engravings on its fin emerges from the turbulent waters below, symbolizing the looming threats and inner turmoil. " "The scene is painted with rich, dark tones and surreal, otherworldly elements, blending fantasy with a dramatic psychological 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()