!pip -q install -U diffusers transformers accelerate

import torch
from diffusers import DiffusionPipeline
import matplotlib.pyplot as plt

print("Loading model...")

model = DiffusionPipeline.from_pretrained(
    "segmind/tiny-sd",
    torch_dtype=torch.float32
)

print("Model loaded successfully!")
print("Generating image...")

image = model(
    "A futuristic smart manufacturing factory with robots",
    num_inference_steps=10
).images[0]

print("Image generated successfully!")

plt.imshow(image)
plt.axis("off")
plt.show()