Update app.py

app.py

@@ -5,6 +5,12 @@ from sklearn.preprocessing import OneHotEncoder, StandardScaler
 from keras.models import load_model
 import matplotlib.pyplot as plt
 from keras.layers import LSTM
+import keras
+from keras import backend as K
+
+@keras.saving.register_keras_serializable()
+def custom_mse(y_true, y_pred):
+    return K.mean(K.square(y_pred - y_true))
 
 def custom_lstm(*args, **kwargs):
     kwargs.pop('time_major', None)  # Remove the unsupported argument