Create app.py

app.py

@@ -0,0 +1,53 @@
+import gradio as gr
+from PIL import Image
+import torch, gc
+from diffusers import StableDiffusionXLInpaintPipeline, StableDiffusionXLImg2ImgPipeline
+
+def inpaint_with_mask(image: Image.Image, mask: Image.Image, prompt: str = "background") -> Image.Image:
+    image = image.resize((1024, 1024))
+    mask = mask.resize((1024, 1024)).convert("L")
+
+    # 🧠 Load Inpainting Pipeline
+    pipe = StableDiffusionXLInpaintPipeline.from_pretrained(
+        "diffusers/stable-diffusion-xl-1.0-inpainting-0.1",
+        torch_dtype=torch.float16,
+        variant="fp16"
+    ).to("cuda")
+
+    # 🖌️ Inpaint
+    result = pipe(prompt=prompt, image=image, mask_image=mask).images[0]
+
+    # 🧹 Unload Inpainting
+    del pipe
+    torch.cuda.empty_cache()
+    gc.collect()
+
+    # 🎨 Load Refiner
+    refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained(
+        "stabilityai/stable-diffusion-xl-refiner-1.0",
+        torch_dtype=torch.float16,
+        variant="fp16"
+    ).to("cuda")
+
+    result = refiner(prompt=prompt, image=result).images[0]
+
+    del refiner
+    torch.cuda.empty_cache()
+    gc.collect()
+
+    return result
+
+# Gradio Interface
+with gr.Blocks() as demo:
+    gr.Markdown("# 🖌️ Inpaint with Stable Diffusion XL")
+    with gr.Row():
+        image_input = gr.Image(label="Original Image", type="pil")
+        mask_input = gr.Image(label="Mask (white = inpaint)", type="pil")
+    prompt_input = gr.Textbox(label="Prompt", value="background")
+    output = gr.Image(label="Result")
+
+    run_btn = gr.Button("Inpaint")
+
+    run_btn.click(fn=inpaint_with_mask, inputs=[image_input, mask_input, prompt_input], outputs=output)
+
+demo.launch()