cover_generator

Generate cover image for ebooks using stable diffusion ai

Open https://www.kaggle.com/code/

Click Settings

Find Accelerator

Select: ✔ GPU (T4)

---

first cell:

!pip uninstall -y protobuf peft
!pip install --upgrade protobuf==4.25.3 peft diffusers transformers accelerate safetensors sentencepiece --quiet

second cell:

import torch
from diffusers import StableDiffusionXLPipeline
from PIL import Image

# -----------------------------------------
# WRITE YOUR SUMMARY HERE
# -----------------------------------------
summary = """
photorealistic ... (prompt)
"""
# -----------------------------------------

prompt = (
    "photorealistic, nice image to be used for ebook cover "
    "Based on this summary:\n\n"
    f"{summary}"
)

print("Loading SDXL model...")
pipe = StableDiffusionXLPipeline.from_pretrained(
    "SG161222/RealVisXL_V4.0", # "stabilityai/sdxl-turbo",
    torch_dtype=torch.float16,
    variant="fp16"
).to("cuda")

print("Generating image...")
image = pipe(
    prompt=prompt,
    height=1024,
    width=1024,
    num_inference_steps=30,
    guidance_scale=7.0
).images[0]

output_path = "/kaggle/working/cover.png"
image.save(output_path)

print("Saved to:", output_path)
image