Update app.py

app.py

@@ -36,6 +36,26 @@ def main():
         
         st.markdown(f"<div style='text-align: center; font-size: 30px;'>Predicted Alzheimer's stage is: <b>{class_names[predicted_class]}</b></div>", unsafe_allow_html=True)
         st.image(uploaded_file, use_column_width=False, width=300)
+    else:
+        # Display four example images if no image is uploaded
+        example_images = ["mild_468_0_4983.jpg", "moderate_2_0_72.jpg", "non_61.jpg", "verymild_37_0_2606.jpg"]
+        col1, col2, col3, col4 = st.columns(4)
+        
+        with col1:
+            image1 = Image.open(example_images[0])
+            st.image(image1, caption="Example 1", use_column_width=True)
+            
+        with col2:
+            image2 = Image.open(example_images[1])
+            st.image(image2, caption="Example 2", use_column_width=True)
+            
+        with col3:
+            image3 = Image.open(example_images[2])
+            st.image(image3, caption="Example 3", use_column_width=True)
+            
+        with col4:
+            image4 = Image.open(example_images[3])
+            st.image(image4, caption="Example 4", use_column_width=True)
 
 if __name__ == '__main__':
     main()