adwdawdawadw

main.py

@@ -61,10 +61,10 @@ async def predict_image(file: UploadFile = File(...)):
         
         # change the mask to base64 and calculate the score
         for i in range(len(results)):
+            mask_area = calculate_mask_area(results[i]["mask"])
             results[i]["mask"] = image_to_base64(results[i]["mask"])
             if results[i]["label"] == "background":
                 continue
-            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