switched to .h5

src/streamlit_app.py

@@ -12,7 +12,7 @@ from tensorflow.keras.applications.mobilenet_v2 import preprocess_input
 # === Load model ===
 @st.cache_resource
 def load_model():
-    model_path = os.path.join("src", "model", "dog_breed_classifier.keras")
+    model_path = os.path.join("src", "model", "dog_breed_classifier.h5")
     model = tf.keras.models.load_model(model_path)
     return model
 

train_model.py

@@ -12,7 +12,7 @@ import matplotlib.pyplot as plt
 
 # === Paths ===
 DATA_DIR = "data/train"
-MODEL_SAVE_PATH = "src/model/dog_breed_classifier.keras"
+MODEL_SAVE_PATH = "src/model/dog_breed_classifier.h5"
 CLASS_NAMES_PATH = "src/model/class_names.json"
 IMG_SIZE = (224, 224)
 BATCH_SIZE = 32