Update app.py

app.py

@@ -4,10 +4,18 @@ import pandas as pd
 from sklearn.preprocessing import OneHotEncoder, StandardScaler
 from keras.models import load_model
 import matplotlib.pyplot as plt
+from keras.layers import LSTM
+
+def custom_lstm(*args, **kwargs):
+    kwargs.pop('time_major', None)  # Remove the unsupported argument
+    return LSTM(*args, **kwargs)
+
+
+
 
 # Load the trained model
 try:
-    loaded_model = load_model('solar_irradiance_model.h5')
+    loaded_model = load_model('solar_irradiance_model.h5', custom_objects={'LSTM': custom_lstm})
 except ValueError as e:
     st.error(f"Error loading model: {e}")