Update app.py

app.py

@@ -37,7 +37,7 @@ controlnet = ControlNetModel.from_pretrained(
     "monster-labs/control_v1p_sd15_qrcode_monster",
     #"DionTimmer/controlnet_qrcode-control_v1p_sd15", 
     torch_dtype=torch.float16
-).to("cuda")
+).to("mps")
 
 pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
     "digiplay/GhostMixV1.2VAE",
@@ -45,7 +45,7 @@ pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
     controlnet = controlnet,
     torch_dtype = torch.float16,
     safety_checker = None,
-).to("cuda")
+).to("mps")
 #pipe.enable_xformers_memory_efficient_attention()
 
 
@@ -113,7 +113,20 @@ def scan_qr_code(image):
 def invert_image(image):
     if image is None:
         return None
-    return ImageOps.invert(image.convert('RGB'))
+    if isinstance(image, np.ndarray):
+        return 255 - image
+    elif isinstance(image, Image.Image):
+        return ImageOps.invert(image.convert('RGB'))
+    else:
+        raise ValueError("Unsupported image type")
+
+def invert_displayed_image(image):
+    if image is None:
+        return None
+    inverted = invert_image(image)
+    if isinstance(inverted, np.ndarray):
+        return Image.fromarray(inverted)
+    return inverted
 
 @spaces.GPU()
 def inference(