Upload 2 files

app.py

@@ -0,0 +1,29 @@
+import gradio as gr
+import torch
+from diffusers import StableDiffusionXLPipeline
+
+model_id = "stabilityai/stable-diffusion-xl-base-1.0"
+
+pipe = StableDiffusionXLPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
+
+def generate(prompt):
+    image = pipe(prompt).images[0]
+    return image
+
+title = "🛋️ Virtual Room Decorator"
+description = "Generate stunning room decor with AI using SDXL. Describe your room style, and get instant visuals."
+
+gr.Interface(
+    fn=generate,
+    inputs=gr.Textbox(lines=2, placeholder="e.g. a cozy living room with wooden floors and indoor plants"),
+    outputs=gr.Image(type="pil"),
+    title=title,
+    description=description,
+    theme="soft",
+    examples=[
+        ["A minimalist bedroom with a large window and warm lighting"],
+        ["A futuristic workspace with sleek furniture and LED lights"],
+        ["A kid's room with toys, a bunk bed, and cartoon wall art"]
+    ]
+).launch()

requirements.txt

@@ -0,0 +1,6 @@
+diffusers==0.34.0
+transformers
+torch
+gradio
+accelerate
+safetensors