Auto-deploy from GitHub: b55422c85fd316d361947d76e66614417b8b49ef

.gitattributes

@@ -33,3 +33,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.jpeg filter=lfs diff=lfs merge=lfs -text
+*.jpg filter=lfs diff=lfs merge=lfs -text
+*.png filter=lfs diff=lfs merge=lfs -text

app.py

@@ -188,8 +188,25 @@ if model_loaded:
             type=["jpg", "jpeg", "png"]
         )
 
+        # Sample images
+        st.subheader("🖼️ Or Try Sample Images")
+        sample_col1, sample_col2 = st.columns(2)
+        with sample_col1:
+            if st.button("Normal Sample", use_container_width=True):
+                st.session_state.sample = "samples/normal_sample.jpeg"
+        with sample_col2:
+            if st.button("Pneumonia Sample", use_container_width=True):
+                st.session_state.sample = "samples/pneumonia_sample.jpeg"
+
+    # Determine which image to use
+    image = None
     if uploaded_file is not None:
         image = Image.open(uploaded_file).convert("RGB")
+        st.session_state.sample = None  # Clear sample when uploading
+    elif "sample" in st.session_state and st.session_state.sample:
+        image = Image.open(st.session_state.sample).convert("RGB")
+
+    if image is not None:
 
         with col1:
             st.image(image, caption="Uploaded X-Ray", use_container_width=True)
@@ -233,6 +250,6 @@ if model_loaded:
 
                 st.warning("**Disclaimer:** For educational purposes only. Consult a healthcare professional.")
 
-# Footer
+# Footers
 st.markdown("---")
 st.markdown("<p style='text-align:center;color:#888;'>Built with PyTorch & Streamlit</p>", unsafe_allow_html=True)