Update app.py

app.py

@@ -5,12 +5,13 @@ import spaces
 import torch
 from diffusers import AutoPipelineForInpainting
 from loguru import logger
-from PIL import Image
+from PIL import Image, ImageChops
 
 SUPPORTED_MODELS = [
     "stabilityai/sdxl-turbo",
     "stabilityai/stable-diffusion-3-medium-diffusers",
     "stabilityai/stable-diffusion-xl-base-1.0",
+    "stable-diffusion-v1-5/stable-diffusion-v1-5",
     "timbrooks/instruct-pix2pix",
 ]
 DEFAULT_MODEL = "stabilityai/stable-diffusion-xl-base-1.0"
@@ -27,7 +28,7 @@ def load_pipeline(model):
 
 
 logger.debug(f"Loading pipeline: {dict(model=model)}")
-pipe = load_pipeline(model).to("cuda")
+pipe = load_pipeline(model).to("cuda" if torch.cuda.is_available() else "mps")
 
 
 @logger.catch(reraise=True)
@@ -46,10 +47,14 @@ def infer(
     )
 
     init_image: Image.Image = image_editor["background"].convert("RGB")
-    mask_image: Image.Image = image_editor["layers"][0].convert("RGB")
 
     # Downscale the image
     init_image.thumbnail((1024, 1024))
+
+    mask_layer = image_editor["layers"][0]
+    mask_image = Image.new("RGBA", mask_layer.size, "white")
+    mask_image = Image.alpha_composite(mask_image, mask_layer).convert("RGB")
+    mask_image = ImageChops.invert(mask_image)
     mask_image.thumbnail((1024, 1024))
 
     additional_args = {