Update app.py

app.py

@@ -57,6 +57,7 @@ def infer(edit_images, prompt, seed=42, randomize_seed=False, width=1024, height
     image = edit_images["background"]
     width, height = calculate_optimal_dimensions(image)
     mask = edit_images["layers"][0]
+    mask = mask.convert('L')
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     image = pipe(