Update app.py

app.py

@@ -64,14 +64,24 @@ def segment(image):
     mask_image *= 255
     mask_image = mask_image.astype(np.uint8)
     mask_image = Image.fromarray(mask_image).convert("L")
-    return mask_image
+
+    #Porcentaje de 0
+    positive_pixels = np.count_nonzero(mask_image)
+    total_pixels = mask_image.size[0] * mask_image.size[1]
+    percentage = (positive_pixels / total_pixels) * 100
+
+     # Calcular los porcentajes de 0 y 1
+    class_0_percentage = 100 - percentage
+    class_1_percentage = percentage
+
+    return mask_image, class_0_percentage, class_1_percentage
 
 if __name__ == "__main__":
     model = build_model(input_shape=(size, size, 1))
     gr.Interface(
         fn=segment,
         inputs="image",
-        outputs=gr.Image(type="pil", label="Breast Cancer Mask"),
+        outputs=[gr.Image(type="pil", label="Breast Cancer Mask"),"number", "number"],
         examples=[["benign(10).png"], ["benign(109).png"]],
         title = '<h1 style="text-align: center;">Breast Cancer Ultrasound Image Segmentation! 💐 </h1>',
         description = """