Update app.py

app.py

@@ -427,10 +427,18 @@ def load_text_model():
 
 @st.cache_resource(show_spinner=False)
 def load_image_model():
-    path  = hf_hub_download(
-        repo_id="Muniba930/image-detector",
-        filename="image_detector_v2.h5"
-    )
+    # 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"
+        )
+    except Exception:
+        path = hf_hub_download(
+            repo_id="syeda-Rija20/image-detector",
+            filename="image_detector_v2.h5"
+        )
     model = tf.keras.models.load_model(path)
     return model