Update app.py

app.py

@@ -41,17 +41,20 @@ st.markdown(
     "This app classifies bacterial morphology into **Cocci**, **Bacilli**, or **Spirilla** using a fine-tuned PyTorch model."
 )
 
-# Example Images
+# Display example images from local files
 st.subheader("Example Images")
-example_images = [
-    "https://huggingface.co/datasets/yolac/BacterialMorphologyClassification/resolve/main/img%20290.jpg",  # Replace with actual paths to example images
+example_image_paths = [
+    "https://huggingface.co/datasets/yolac/BacterialMorphologyClassification/resolve/main/img%20290.jpg",  
     "https://huggingface.co/datasets/yolac/BacterialMorphologyClassification/resolve/main/img%20565.jpg",
     "https://huggingface.co/datasets/yolac/BacterialMorphologyClassification/resolve/main/img%208.jpg"
 ]
 
-for img_path in example_images:
-    img = Image.open(img_path).convert("RGB")
-    st.image(img, caption=f"Example Image: {img_path}", use_column_width=True)
+for img_path in example_image_paths:
+    try:
+        img = Image.open(img_path).convert("RGB")
+        st.image(img, caption=f"Example Image: {img_path}", use_column_width=True)
+    except FileNotFoundError:
+        st.error(f"Example image {img_path} not found. Please ensure the image exists in the app directory.")
 
 # File Upload
 uploaded_file = st.file_uploader("Upload a bacterial image for classification:", type=["jpg", "jpeg", "png"])