Shaikat01 commited on
Commit
7e6a7e0
·
verified ·
1 Parent(s): f19cf54

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -30,9 +30,8 @@ def load_models():
30
  prophet_model = pickle.load(f)
31
 
32
  # LSTM + scaler
33
- lstm_model = load_model('lstm_model') # use SavedModel folder
34
- with open('lstm_scaler.pkl', 'rb') as f:
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: