Update app.py

app.py

@@ -24,7 +24,8 @@ tf.get_logger().setLevel('ERROR')
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
 
 # ========== CUSTOM LAYERS DEFINITIONS ==========
-@keras.saving.register_keras_serializable()
+# Fixed decorator - using tf.keras.utils.register_keras_serializable
+@tf.keras.utils.register_keras_serializable()
 class RepeatChannels(keras.layers.Layer):
     """Converts single channel (depth) to 3 channels for RGB models"""
     def __init__(self, **kwargs):
@@ -38,7 +39,7 @@ class RepeatChannels(keras.layers.Layer):
         return config
 
 # Add any other custom layers your models might need
-@keras.saving.register_keras_serializable()
+@tf.keras.utils.register_keras_serializable()
 class CustomLayer(keras.layers.Layer):
     """Template for additional custom layers if needed"""
     def __init__(self, **kwargs):