Update app.py

app.py

@@ -7,7 +7,7 @@ import base64
 import cv2
 import numpy as np
 import matplotlib.pyplot as plt
-
+ 
 H = 256
 W = 256
 
@@ -61,6 +61,23 @@ def show_image(image, title="Image"):
     plt.axis('off')
     st.pyplot()
 
+# Function to download sample images
+def download_sample_images():
+    sample_images_folder = "sample_images"
+    sample_images = os.listdir(sample_images_folder)
+
+    for image_name in sample_images:
+        image_path = os.path.join(sample_images_folder, image_name)
+        with open(image_path, "rb") as f:
+            image_bytes = f.read()
+            st.download_button(
+                label=f"Download {image_name}",
+                data=image_bytes,
+                key=f"download_{image_name}",
+                file_name=image_name,
+                mime="image/jpeg",
+            )
+
 # Streamlit app
 def main():
     st.title("Brain Tumor Segmentation App")