v3

app.py

@@ -0,0 +1,16 @@
+import gradio as gr
+from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
+import torch
+
+model_id = "stabilityai/stable-diffusion-2-1-base"
+
+scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
+pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)
+pipe = pipe.cpu()  # Use CPU instead of GPU
+
+def generate_image(prompt):
+    image = pipe(prompt).images[0]
+    return image  # Gradio will handle the conversion to an image
+
+iface = gr.Interface(fn=generate_image, inputs=gr.inputs.Textbox(), outputs='image')
+iface.launch()

requirements.txt

@@ -0,0 +1,6 @@
+transformers
+diffusers 
+torch
+gradio
+pillow
+accelerate