I made a code for ai image

[markhajitov-lab]

from diffusers import StableDiffusionPipeline
import torch

pipe = StableDiffusionPipeline.from_pretrained(
"runwayml/stable-diffusion-v1-5",
torch_dtype=torch.float16
).to("cuda")

prompt = "a futuristic city, neon lights, ultra detailed"

image = pipe(prompt).images[0]
image.save("output.png")