| from diffusers import StableDiffusionPipeline | |
| def generate_image(prompt): | |
| model = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5") | |
| model.to("cuda") # Use GPU for faster generation | |
| image = model(prompt).images[0] | |
| image.save("output.png") | |
| return "output.png" | |
| if __name__ == "__main__": | |
| prompt = "A friendly person saying 'How are you?'" | |
| print("Generated Image Path:", generate_image(prompt)) | |