mboh ah

main.py

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