Update app.py

app.py

@@ -461,28 +461,13 @@ def load_text_model():
     #     except Exception as e:
     #         raise RuntimeError(f"Failed to load image model: {e}")
 
-    # return model
 @st.cache_resource(show_spinner=False)
 def load_image_model():
-    # ── Patch incompatible Keras layers BEFORE any loading ──
-    try:
-        import keras
-        from keras.layers import RandomFlip, RandomRotation, RandomZoom, RandomBrightness, RandomContrast
-
-        for cls in [RandomFlip, RandomRotation, RandomZoom, RandomBrightness, RandomContrast]:
-            _orig = cls.__init__
-            def _patched(self, *args, _orig=_orig, **kwargs):
-                kwargs.pop("data_format", None)
-                _orig(self, *args, **kwargs)
-            cls.__init__ = _patched
-    except Exception:
-        pass  # If patching fails, try loading anyway
-
     filenames_to_try = [
-        "image_detector_v2.keras",
+        "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:
@@ -500,27 +485,10 @@ def load_image_model():
             "Check that 'syeda-Rija20/image-detector' is public and the file exists."
         )
 
-    try:
-        import keras
-        model = keras.saving.load_model(path, compile=False)
-        return model
-    except Exception as e1:
-        try:
-            import tensorflow as tf
-            model = tf.keras.models.load_model(path, compile=False)
-            return model
-        except Exception as e2:
-            raise RuntimeError(
-                f"Failed to load image model.\n"
-                f"keras attempt: {e1}\n"
-                f"tf.keras attempt: {e2}"
-            )
-
-       
-
-          
-
-
+    import keras
+    model = keras.saving.load_model(path, compile=False)
+    return model
+      
 # ─────────────────────────────────────────────
 #  PREDICTION FUNCTIONS
 # ─────────────────────────────────────────────