runtime error

Exit code: 1. Reason: 5-06-09 11:01:09.512439: I tensorflow/core/platform/cpu_feature_guard.cc:210] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations. To enable the following instructions: AVX2 AVX512F AVX512_VNNI FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags. Traceback (most recent call last): File "/home/user/app/app.py", line 12, in <module> model = tf.keras.models.load_model("tft_model.keras", compile=False) File "/usr/local/lib/python3.10/site-packages/keras/src/saving/saving_api.py", line 189, in load_model return saving_lib.load_model( File "/usr/local/lib/python3.10/site-packages/keras/src/saving/saving_lib.py", line 370, in load_model return _load_model_from_fileobj( File "/usr/local/lib/python3.10/site-packages/keras/src/saving/saving_lib.py", line 447, in _load_model_from_fileobj model = _model_from_config( File "/usr/local/lib/python3.10/site-packages/keras/src/saving/saving_lib.py", line 436, in _model_from_config model = deserialize_keras_object( File "/usr/local/lib/python3.10/site-packages/keras/src/saving/serialization_lib.py", line 694, in deserialize_keras_object cls = _retrieve_class_or_fn( File "/usr/local/lib/python3.10/site-packages/keras/src/saving/serialization_lib.py", line 810, in _retrieve_class_or_fn raise TypeError( TypeError: Could not locate class 'SimplifiedTFT_Iter3'. Make sure custom classes are decorated with `@keras.saving.register_keras_serializable()`. Full object config: {'module': None, 'class_name': 'SimplifiedTFT_Iter3', 'config': {'name': 'simple_tft_iter3', 'trainable': True, 'dtype': {'module': 'keras', 'class_name': 'DTypePolicy', 'config': {'name': 'float32'}, 'registered_name': None}}, 'registered_name': 'SimplifiedTFT_Iter3', 'build_config': {'input_shape': {'past_target': [32, 24, 1], 'past_observed_covariates': [32, 24, 46], 'static_features': [32, 2], 'past_known_covariates': [32, 24, 4], 'future_known_covariates': [32, 1, 4]}}, 'compile_config': None}

Container logs: