app.py

import gradio as gr
from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
import torch
import random

# No authentication needed for SD v2
model_id = "stabilityai/stable-diffusion-2-1-base"

# Initialize the model
scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
pipe = StableDiffusionPipeline.from_pretrained(
    model_id,
    scheduler=scheduler,
    torch_dtype=torch.float32  # Works on CPU
).to("cpu")

def generate_image(prompt):
    seed = random.randint(0, 1000000)  # Random seed
    generator = torch.Generator("cpu").manual_seed(seed)
    image = pipe(prompt, generator=generator).images[0]
    return image

# Create interface
gr.Interface(
    fn=generate_image,
    inputs=gr.Textbox(label="Type anything (e.g., 'dog')"),
    outputs=gr.Image(label="Generated Image"),
    title="Free AI Image Generator"
).launch()