Update app.py

app.py

@@ -31,25 +31,8 @@ custom_style = """
 
 st.markdown(custom_style, unsafe_allow_html=True)
 
-def preprocess_image(img):
-    img = img.resize(img_size)
-    img_array = image.img_to_array(img)
-    img_array = img_array / 255.0
-    img_array = np.expand_dims(img_array, axis=0)
-    return img_array
 
-def predict_tumor_segmentation(img):
-    img_array = preprocess_image(img)
-    segmentation_mask = model.predict(img_array)
-    segmentation_mask = np.squeeze(segmentation_mask, axis=0)
-    return segmentation_mask
 
-def display_segmentation_result(original_image, segmentation_mask):
-    original_image = np.array(original_image)
-    segmentation_mask = (segmentation_mask > 0.5).astype(np.uint8) * 255
-    segmentation_mask = Image.fromarray(segmentation_mask, 'L')
-
-    st.image([original_image, segmentation_mask], caption=['Original Image', 'Segmentation Mask'], use_column_width=True)
 
 def main():
     st.title("Brain Tumor Segmentation App")
@@ -62,11 +45,5 @@ def main():
 
         st.markdown("## Tumor Segmentation Result")
 
-        # Perform segmentation
-        segmentation_mask = predict_tumor_segmentation(original_image)
-
-        # Display the segmentation result
-        display_segmentation_result(original_image, segmentation_mask)
-
 if __name__ == "__main__":
     main()