Update app.py

app.py

@@ -81,7 +81,11 @@ if __name__ == "__main__":
     gr.Interface(
         fn=segment,
         inputs="image",
-        outputs=["image","number", "number"],
+        outputs=[
+            gr.Image(type="pil", label="Breast Cancer Mask"),
+            gr.Number(label="Class 0 Percentage"),
+            gr.Number(label="Class 1 Percentage")
+        ],
         examples=[["benign(10).png"], ["benign(109).png"]],
         title = '<h1 style="text-align: center;">Breast Cancer Ultrasound Image Segmentation! 💐 </h1>',
         description = """