Update app.py

app.py

@@ -5,20 +5,22 @@ from PIL import Image
 
 MODEL_ID = "runwayml/stable-diffusion-v1-5"
 
+# ✅ FORCE CPU
+device = "cpu"
+dtype = torch.float32
+
 pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
     MODEL_ID,
-    torch_dtype=torch.float16
-).to("cuda")
-
-pipe.safety_checker = None  # optional, speeds things up
-
-def redesign_room(
-    image,
-    style,
-    colors,
-    lighting,
-    strength
-):
+    torch_dtype=dtype
+)
+
+pipe = pipe.to(device)
+
+# Disable safety checker safely
+pipe.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
+
+
+def redesign_room(image, style, colors, lighting, strength):
     prompt = f"""
 Redesign the interior of this room while keeping the same layout,
 walls, windows, doors, and camera angle.
@@ -38,31 +40,25 @@ interior photography, high detail.
         image=image,
         strength=strength,
         guidance_scale=7.5,
-        num_inference_steps=30
+        num_inference_steps=25
     ).images[0]
 
     return result
 
-with gr.Blocks(title="Room Redesign AI") as demo:
-    gr.Markdown("## 🏠 AI Room Redesign (Image-to-Image)")
-
-    with gr.Row():
-        image_input = gr.Image(label="Upload Room Image", type="pil")
-
-    with gr.Row():
-        style = gr.Dropdown(
-            ["Modern", "Luxury", "Scandinavian", "Minimal", "Japanese"],
-            value="Modern",
-            label="Interior Style"
-        )
-        colors = gr.Textbox(
-            value="white, beige, wood",
-            label="Color Palette"
-        )
-        lighting = gr.Textbox(
-            value="warm ambient lighting",
-            label="Lighting"
-        )
+
+with gr.Blocks(title="Room Redesign AI (CPU)") as demo:
+    gr.Markdown("## 🏠 AI Room Redesign (CPU Mode)")
+
+    image_input = gr.Image(label="Upload Room Image", type="pil")
+
+    style = gr.Dropdown(
+        ["Modern", "Luxury", "Scandinavian", "Minimal", "Japanese"],
+        value="Modern",
+        label="Interior Style"
+    )
+
+    colors = gr.Textbox(value="white, beige, wood", label="Color Palette")
+    lighting = gr.Textbox(value="warm ambient lighting", label="Lighting")
 
     strength = gr.Slider(
         minimum=0.2,
@@ -75,7 +71,6 @@ with gr.Blocks(title="Room Redesign AI") as demo:
     output = gr.Image(label="Redesigned Room")
 
     btn = gr.Button("Redesign Room")
-
     btn.click(
         redesign_room,
         inputs=[image_input, style, colors, lighting, strength],