22

app.py

@@ -148,8 +148,8 @@ def inference(
     # control_image = control_image.resize((width, height))
     # control_image_torch = get_image_latent(control_image, sample_size=sample_size)[:, :, 0]
 
-    # mask_image = edit_dict['layers'][0]
-    mask_image = edit_dict['composite']
+    mask_image = edit_dict['layers'][0]
+    # mask_image = edit_dict['composite']
     if mask_image is not None:
         mask_image = get_image_latent(mask_image, sample_size=sample_size)[:, :1, 0]
     else:
@@ -209,6 +209,7 @@ with gr.Blocks() as demo:
                     sources=['upload', 'clipboard'], 
                     image_mode='RGB', 
                     type="pil",
+                    image_mode= "L",
                     label="Mask Image"
                 )