Update app.py
Browse files
app.py
CHANGED
|
@@ -46,7 +46,7 @@ def classify_image_and_stats(image_input):
|
|
| 46 |
total_area = IMG_HEIGHT * IMG_WIDTH
|
| 47 |
tumor_ratio = tumor_area / total_area
|
| 48 |
|
| 49 |
-
tumor_label = "Tumor Detected" if tumor_ratio > 0.
|
| 50 |
|
| 51 |
overlay = np.array(img)
|
| 52 |
red_mask = np.zeros_like(overlay)
|
|
@@ -125,7 +125,7 @@ inputs = [
|
|
| 125 |
|
| 126 |
outputs = [
|
| 127 |
gr.Image(type="numpy", label="Overlay: Brain MRI + Tumor Mask"),
|
| 128 |
-
gr.Textbox(label="Doctor's Explanation
|
| 129 |
]
|
| 130 |
|
| 131 |
if __name__ == "__main__":
|
|
|
|
| 46 |
total_area = IMG_HEIGHT * IMG_WIDTH
|
| 47 |
tumor_ratio = tumor_area / total_area
|
| 48 |
|
| 49 |
+
tumor_label = "Tumor Detected" if tumor_ratio > 0.0035 else "No Tumor Detected"
|
| 50 |
|
| 51 |
overlay = np.array(img)
|
| 52 |
red_mask = np.zeros_like(overlay)
|
|
|
|
| 125 |
|
| 126 |
outputs = [
|
| 127 |
gr.Image(type="numpy", label="Overlay: Brain MRI + Tumor Mask"),
|
| 128 |
+
gr.Textbox(label="Doctor's Explanation")
|
| 129 |
]
|
| 130 |
|
| 131 |
if __name__ == "__main__":
|