Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -29,15 +29,19 @@ def load_models():
|
|
| 29 |
with open('prophet_model.pkl', 'rb') as f:
|
| 30 |
prophet_model = pickle.load(f)
|
| 31 |
|
| 32 |
-
# LSTM
|
| 33 |
-
from tensorflow.keras.models import load_model
|
| 34 |
lstm_model = load_model('lstm_model.keras')
|
| 35 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 36 |
return arima_model, prophet_model, lstm_model, scaler
|
| 37 |
except Exception as e:
|
| 38 |
print(f"Error loading models: {e}")
|
| 39 |
return None, None, None, None
|
| 40 |
|
|
|
|
| 41 |
arima_model, prophet_model, lstm_model, scaler = load_models()
|
| 42 |
SEQ_LENGTH = 60
|
| 43 |
|
|
|
|
| 29 |
with open('prophet_model.pkl', 'rb') as f:
|
| 30 |
prophet_model = pickle.load(f)
|
| 31 |
|
| 32 |
+
# LSTM model
|
|
|
|
| 33 |
lstm_model = load_model('lstm_model.keras')
|
| 34 |
|
| 35 |
+
# LSTM scaler
|
| 36 |
+
with open('lstm_scaler.pkl', 'rb') as f:
|
| 37 |
+
scaler = pickle.load(f)
|
| 38 |
+
|
| 39 |
return arima_model, prophet_model, lstm_model, scaler
|
| 40 |
except Exception as e:
|
| 41 |
print(f"Error loading models: {e}")
|
| 42 |
return None, None, None, None
|
| 43 |
|
| 44 |
+
|
| 45 |
arima_model, prophet_model, lstm_model, scaler = load_models()
|
| 46 |
SEQ_LENGTH = 60
|
| 47 |
|