Upload app.py

app.py

@@ -0,0 +1,30 @@
+import torch
+import gradio as gr
+from diffusers import StableDiffusionPipeline
+from diffusers import AutoPipelineForText2Image
+
+pipe = AutoPipelineForText2Image.from_pretrained(
+    "stabilityai/sdxl-turbo",
+    torch_dtype=torch.float16,
+    variant="fp16"
+).to("cpu")
+
+
+# Disable safety checker (optional, to avoid empty images)
+pipe.safety_checker = lambda images, **kwargs: (images, False)
+
+# Image generation function
+def generate_image(prompt):
+    result = pipe(prompt)
+    return result.images[0]
+
+# Gradio UI
+iface = gr.Interface(
+    fn=generate_image,
+    inputs=gr.Textbox(label="Enter your prompt"),
+    outputs=gr.Image(label="Generated Image"),
+    title="Speech2Image 💖",
+    description="Enter a sentence and get a beautiful image ✨"
+)
+
+iface.launch()