haydpw commited on
Commit
d3bf65a
1 Parent(s): 9d7fc49

change mask colors

Browse files
Files changed (1) hide show
  1. main.py +3 -3
main.py CHANGED
@@ -65,9 +65,9 @@ async def predict_image(file: UploadFile = File(...)):
65
  background_area = 0
66
 
67
  mask_dict = {
68
- 'acne': (255, 0, 0),
69
- 'dry': (0, 255, 0),
70
- 'oily': (0,0,255),
71
  'background': (255,255,255)
72
  }
73
  # change the mask to base64 and calculate the score
 
65
  background_area = 0
66
 
67
  mask_dict = {
68
+ 'acne': (245, 177, 177),
69
+ 'dry': (208, 181, 166),
70
+ 'oily': (240, 230, 214),
71
  'background': (255,255,255)
72
  }
73
  # change the mask to base64 and calculate the score