Update app.py

app.py

@@ -10,7 +10,7 @@ from scipy.spatial import cKDTree
 
 # ── Config ───────────────────────────────────────────────────
 MODEL_REPO = "Unearthed/artifact-classifier"
-MODEL_FILE = "lithic_classifier0531.h5"
+MODEL_FILE = "lithic_classifier0531.keras"
 IMG_SIZE   = (224, 224)
 
 CONF_THRESHOLD        = 0.55
@@ -39,6 +39,7 @@ def display_category(cls_name):
 print("Loading model...")
 model_path = hf_hub_download(repo_id=MODEL_REPO, filename=MODEL_FILE)
 import keras
+keras.config.enable_unsafe_deserialization()
 model = tf.keras.models.load_model(model_path, compile=False)
 print("Model ready!")