Update app.py

app.py

@@ -157,7 +157,11 @@ model = None
 try:
     if os.path.exists(MODEL_PATH):
         logger.info("Loading model from %s", MODEL_PATH)
-        model = tf.keras.models.load_model(MODEL_PATH)
+        try:
+            model = tf.keras.models.load_model(MODEL_PATH, compile=False)
+        except Exception as e:
+            logger.error("Failed to deserialize model: %s", str(e))
+            raise ValueError(f"Model deserialization error: {str(e)}. Please ensure the model is compatible with TensorFlow 2.15.0 or re-save it.")
     else:
         logger.error("Model file %s not found", MODEL_PATH)
         raise FileNotFoundError(f"Model file {MODEL_PATH} not found")
@@ -337,7 +341,7 @@ def home():
         return render_template("form.html", history_plot=None, result={
             "prediction": "Error",
             "confidence": "N/A",
-            "message": "Failed to load page. Please try again.",
+            "message": f"Failed to load page: {str(e)}",
             "email_status": "N/A"
         })
 
@@ -438,6 +442,8 @@ def email_report(scan_id):
 @app.route("/predict", methods=["POST"])
 def predict():
     try:
+        if not model:
+            raise ValueError("Model not loaded, cannot make predictions")
         if "image" not in request.files:
             raise ValueError("No image uploaded.")
         image = request.files["image"]