Create app.py

app.py

@@ -0,0 +1,55 @@
+import torch
+import gradio as gr
+from diffusers import StableDiffusionPipeline
+
+model_id = "runwayml/stable-diffusion-v1-5"
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
+pipe = StableDiffusionPipeline.from_pretrained(
+    model_id,
+    torch_dtype=torch.float16 if device == "cuda" else torch.float32
+)
+pipe = pipe.to(device)
+
+styles = {
+    "Realistic": "highly detailed, realistic, 4k",
+    "Cartoon": "cartoon style, colorful",
+    "Anime": "anime style",
+    "Sketch": "pencil sketch"
+}
+
+def generate_image(prompt, style, steps, guidance, seed):
+    if not prompt.strip():
+        return None, "⚠️ Please enter a prompt."
+
+    styled_prompt = prompt + ", " + styles.get(style, "")
+    generator = torch.manual_seed(int(seed)) if seed != 0 else None
+
+    image = pipe(
+        styled_prompt,
+        num_inference_steps=int(steps),
+        guidance_scale=float(guidance),
+        generator=generator
+    ).images[0]
+
+    return image, "✅ Done"
+
+with gr.Blocks() as app:
+    gr.Markdown("# 🎨 Text-to-Image Generator")
+    gr.Markdown("Powered by Stable Diffusion")
+
+    prompt = gr.Textbox(label="Prompt")
+    style = gr.Dropdown(["Realistic", "Cartoon", "Anime", "Sketch"], value="Realistic")
+
+    steps = gr.Slider(10, 50, value=25)
+    guidance = gr.Slider(5.0, 15.0, value=7.5)
+    seed = gr.Number(value=0)
+
+    btn = gr.Button("Generate")
+
+    image = gr.Image()
+    status = gr.Textbox()
+
+    btn.click(generate_image, [prompt, style, steps, guidance, seed], [image, status])
+
+app.launch()