File size: 579 Bytes
a34c2f2 70b14dc | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | from diffusers import AutoPipelineForText2Image
import torch
import gradio as gr
# Load a CPU-friendly model (works without GPU)
pipe = AutoPipelineForText2Image.from_pretrained(
"stabilityai/sd-turbo",
torch_dtype=torch.float32
).to("cpu")
# Image generation function
def generate_image(prompt):
image = pipe(prompt, num_inference_steps=6).images[0]
return image
# Gradio UI
gr.Interface(
fn=generate_image,
inputs=gr.Textbox(label="Enter a prompt"),
outputs=gr.Image(label="Generated Image"),
title="CPU Image Generator (SD-Turbo)"
).launch() |