Update utils/model_loader.py

utils/model_loader.py

@@ -57,7 +57,18 @@ def load_pytorch_model(build_fn, weights_path, device):
 def load_keras_model(weights_path):
     if not os.path.exists(weights_path):
         raise FileNotFoundError(f"Model not found: {weights_path}")
-    return tf.keras.models.load_model(weights_path, compile=False)
+    try:
+        return tf.keras.models.load_model(weights_path, compile=False)
+    except Exception:
+        import h5py
+        with h5py.File(weights_path, 'r') as f:
+            model_config = f.attrs.get('model_config')
+        model = tf.keras.models.model_from_json(
+            model_config,
+            custom_objects=None
+        )
+        model.load_weights(weights_path)
+        return model
 
 def load_all_models():
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")