subek commited on
Commit
9441e59
·
verified ·
1 Parent(s): e0d9030

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -77,16 +77,19 @@ def main():
77
 
78
  # Display original image
79
  st.subheader("Original Image")
80
- show_image(original_image, title="Original Image")
 
81
 
82
  # Display mask
83
  st.subheader("Mask")
84
- show_image(mask, title="Mask")
 
85
 
86
  # Display segmented image
87
  st.subheader("Segmented Brain Tumor")
88
- show_image(segmented_image, title="Segmented Brain Tumor")
89
-
 
90
  if __name__ == "__main__":
91
  main()
92
 
 
77
 
78
  # Display original image
79
  st.subheader("Original Image")
80
+ fig_original = show_image(original_image, title="Original Image")
81
+ st.pyplot(fig_original)
82
 
83
  # Display mask
84
  st.subheader("Mask")
85
+ fig_mask = show_image(mask, title="Mask")
86
+ st.pyplot(fig_mask)
87
 
88
  # Display segmented image
89
  st.subheader("Segmented Brain Tumor")
90
+ fig_segmented = show_image(segmented_image, title="Segmented Brain Tumor")
91
+ st.pyplot(fig_segmented)
92
+
93
  if __name__ == "__main__":
94
  main()
95