app.py

import torch
from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler

# Load the model with torch.float32 for CPU
pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float32)
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)  # Faster scheduler
pipe = pipe.to("cpu")

# Optimized function for faster image generation
def generate_image(prompt):
    image = pipe(prompt, num_inference_steps=15, guidance_scale=7.0).images[0]  # Reduced steps
    image.save("output.png")
    print("Image saved!")

generate_image("a photo of an astronaut riding a horse on Mars")