Shaikat01 commited on
Commit
7fc71cd
·
verified ·
1 Parent(s): 537177c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
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 + 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:
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