Update app.py

app.py

@@ -7,7 +7,7 @@ import os
 # Load the saved model
 @st.cache_resource
 def load_model():
-    model = tf.keras.models.load_model("pneumonia_cnn_model.keras")  # Use relative path for deployment
+    model = tf.keras.models.load_model("pneumonia_cnn_model.keras") 
     return model
 
 model = load_model()
@@ -19,8 +19,8 @@ st.markdown("Upload your own X-ray or try one of the sample images below.")
 # === Sample Image Section ===
 sample_images = {
     "Choose a sample image": None,
-    "🧍 Normal Sample": "normal.jpg",
-    "🤒 Pneumonia Sample": "pneumonia.jpg"
+    "🧍 Normal Sample": "normal.jpeg",
+    "🤒 Pneumonia Sample": "pneumonia.jpeg"
 }
 
 selected_sample = st.selectbox("📂 Select a sample image", list(sample_images.keys()))