coba deh

main.py

@@ -64,7 +64,7 @@ async def predict_image(file: UploadFile = File(...)):
             results[i]["mask"] = image_to_base64(results[i]["mask"])
             if results[i]["label"] == "background":
                 continue
-            mask_area = calculate_mask_area(results[i]["mask"])
+            mask_area = calculate_mask_area(results[i]["mask"].convert("L"))
             print(f"{results[i]['label']} area: {mask_area}")
             score = mask_area / (500 * 500 - background_area)
             results[i]["score"] = score