Update pages/CNN-Model.py
Browse files- pages/CNN-Model.py +1 -0
pages/CNN-Model.py
CHANGED
|
@@ -94,6 +94,7 @@ if uploaded_cell_image and uploaded_cytoplasm_mask and uploaded_nuclei_mask:
|
|
| 94 |
color_coded_image[:, :, 0] = nuclei_mask # Red for nuclei
|
| 95 |
|
| 96 |
plt.imshow(color_coded_image)
|
|
|
|
| 97 |
plt.axis('off')
|
| 98 |
st.pyplot(plt)
|
| 99 |
|
|
|
|
| 94 |
color_coded_image[:, :, 0] = nuclei_mask # Red for nuclei
|
| 95 |
|
| 96 |
plt.imshow(color_coded_image)
|
| 97 |
+
plt.title(f"Predicted: {predicted_label}")
|
| 98 |
plt.axis('off')
|
| 99 |
st.pyplot(plt)
|
| 100 |
|