Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -30,9 +30,8 @@ def load_models():
|
|
| 30 |
prophet_model = pickle.load(f)
|
| 31 |
|
| 32 |
# LSTM + scaler
|
| 33 |
-
|
| 34 |
-
|
| 35 |
-
scaler = pickle.load(f)
|
| 36 |
|
| 37 |
return arima_model, prophet_model, lstm_model, scaler
|
| 38 |
except Exception as e:
|
|
|
|
| 30 |
prophet_model = pickle.load(f)
|
| 31 |
|
| 32 |
# LSTM + scaler
|
| 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:
|