Update app.py

app.py

@@ -463,30 +463,23 @@ def load_text_model():
 
 @st.cache_resource(show_spinner=False)
 def load_image_model():
-    filenames_to_try = [
-        "image_detector_clean.keras",   # ← clean model first
-        "image_detector_v2.keras",      # ← fallbacks
-        "image_detector_v2.h5",
-    ]
-    path = None
-    for fname in filenames_to_try:
-        try:
-            path = hf_hub_download(
-                repo_id="syeda-Rija20/image-detector",
-                filename=fname
-            )
-            break
-        except Exception:
-            continue
-
-    if path is None:
-        raise RuntimeError(
-            "Could not download image model from HuggingFace Hub. "
-            "Check that 'syeda-Rija20/image-detector' is public and the file exists."
-        )
-
     import keras
-    model = keras.saving.load_model(path, compile=False)
+    import json
+
+    config_path = hf_hub_download(
+        repo_id="syeda-Rija20/image-detector",
+        filename="model_config.json"
+    )
+    weights_path = hf_hub_download(
+        repo_id="syeda-Rija20/image-detector",
+        filename="image_detector_clean.weights.h5"
+    )
+
+    with open(config_path) as f:
+        config = f.read()
+
+    model = keras.models.model_from_json(config)
+    model.load_weights(weights_path)
     return model
       
 # ─────────────────────────────────────────────