Update app.py

app.py

@@ -79,7 +79,7 @@ def inpaint_hair(image, prompt):
     
     # Convert to PIL image for the inpainting pipeline
     image_pil = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
-    mask_pil = Image.fromarray(cv2.cvtColor(mask, cv2.COLOR_RGBA2RGB))
+    mask_pil = Image.fromarray(mask[:, :, :3])
 
     # Prepare the inpainting condition
     image_np = np.array(image_pil).astype(np.float32) / 255.0