Update app.py

app.py

@@ -427,19 +427,40 @@ def load_text_model():
 
 @st.cache_resource(show_spinner=False)
 def load_image_model():
-    # Try .keras first, fall back to .h5
     import os
-    try:
-        path = hf_hub_download(
-            repo_id="syeda-Rija20/image-detector",
-            filename="image_detector_v2.keras"
+    import keras  # Keras 3 explicit import
+
+    filenames_to_try = [
+        "image_detector_v2.keras",
+        "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."
         )
+
+    # Try keras 3 native load first, then tf.keras fallback
+    try:
+        model = keras.saving.load_model(path, compile=False)
     except Exception:
-        path = hf_hub_download(
-            repo_id="syeda-Rija20/image-detector",
-            filename="image_detector_v2.h5"
-        )
-    model = tf.keras.models.load_model(path)
+        try:
+            model = tf.keras.models.load_model(path, compile=False)
+        except Exception as e:
+            raise RuntimeError(f"Failed to load image model: {e}")
+
     return model