fix color

app.py

@@ -32,8 +32,8 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
 bfl_repo="black-forest-labs/FLUX.1-dev"
 
-BG_COLOR = (0, 0, 0) # black
-MASK_COLOR = (255, 255, 255) # white
+BG_COLOR = (255, 255, 255) # white
+MASK_COLOR = (0, 0 , 0) # black
 
 def maskHead(input):
   base_options = python.BaseOptions(model_asset_path='selfie_multiclass_256x256.tflite')
@@ -123,7 +123,7 @@ def execute(image, prompt):
   for image in range(len(imgs)):
     current_img = imgs[image]
     cv2.imwrite('base_image.jpg', current_img)
-    cv2.imwrite("mask.jpg", cv2.bitwise_not(maskHead('base_image.jpg')))
+    cv2.imwrite("mask.jpg", maskHead('base_image.jpg'))
     
     im = Image.open('base_image.jpg')
     mask = Image.open('mask.jpg')
@@ -142,7 +142,6 @@ def execute(image, prompt):
         ).images[0]
     response.append(result)
 
-
   return response
 
 iface = gr.Interface(