final config

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import tensorflow as tf
+import keras
 import numpy as np
 import pandas as pd
 import os
@@ -28,13 +29,24 @@ def load_resources():
     if os.path.exists(MODEL_PATH):
         try:
             # Load the full multi-output model
-            model = tf.keras.models.load_model(MODEL_PATH, compile=False)
+            model = keras.saving.load_model(
+                MODEL_PATH,
+                compile=False,
+                safe_mode=False,
+                custom_objects={
+                    "Dense": DenseCompat,
+                    "keras.layers.Dense": DenseCompat,
+                },
+            )
         except Exception:
             try:
                 model = tf.keras.models.load_model(
                     MODEL_PATH,
                     compile=False,
-                    custom_objects={"Dense": DenseCompat},
+                    custom_objects={
+                        "Dense": DenseCompat,
+                        "keras.layers.Dense": DenseCompat,
+                    },
                 )
             except Exception as inner_e:
                 load_error = str(inner_e)