change mask colors
Browse files
main.py
CHANGED
@@ -65,9 +65,9 @@ async def predict_image(file: UploadFile = File(...)):
|
|
65 |
background_area = 0
|
66 |
|
67 |
mask_dict = {
|
68 |
-
'acne': (
|
69 |
-
'dry': (
|
70 |
-
'oily': (
|
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
|