Upload 2 files

app.py

@@ -0,0 +1,29 @@
+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()

requirements.txt

@@ -0,0 +1,5 @@
+gradio>=3.50
+diffusers>=0.24.0
+transformers>=4.35.0
+torch>=2.1.0
+accelerate>=0.25.0