Update app.py

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 from PIL import Image
 import os
 import torch
-from diffusers import StableDiffusionXLInpaintPipeline
+from diffusers import StableDiffusionInpaintPipeline
 from utils.planner import extract_scene_plan  # 🧠 Brain Layer
 
 # ----------------------------
@@ -16,11 +16,12 @@ dtype = torch.float16 if device == "cuda" else torch.float32
 # ----------------------------
 # 📦 Load Inpainting Model
 # ----------------------------
-pipe = StableDiffusionXLInpaintPipeline.from_pretrained(
-    "diffusers/stable-diffusion-xl-1.0-inpainting",
-    torch_dtype=dtype,
-    variant="fp16" if device == "cuda" else None
-).to(device)
+
+
+pipe = StableDiffusionInpaintPipeline.from_pretrained(
+    "runwayml/stable-diffusion-inpainting",
+    torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
+).to("cuda" if torch.cuda.is_available() else "cpu")
 
 pipe.enable_attention_slicing()
 pipe.enable_model_cpu_offload()
@@ -90,3 +91,6 @@ with gr.Blocks() as demo:
 
 demo.launch()
 
+
+
+