| from diffusers import StableDiffusionXLPipeline | |
| import torch | |
| from PIL import Image | |
| import os | |
| pipe = StableDiffusionXLPipeline.from_pretrained( | |
| "stabilityai/sdxl-turbo" | |
| ).to("cpu") | |
| pipe.safety_checker = None # Optional | |
| def generate_scene_image(prompt, idx): | |
| image = pipe(prompt).images[0] | |
| out_path = f"assets/images/scene_{idx}.png" | |
| image.save(out_path) | |
| return out_path |