AbdullahImran commited on
Commit
5db8369
·
verified ·
1 Parent(s): 0697066

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -17,6 +17,8 @@ scaler_X = joblib.load("scaler_X.pkl")
17
  scaler_y = joblib.load("scaler_y.pkl")
18
  lstm_model = load_model("lstm_revenue_model.keras")
19
 
 
 
20
  # Set matplotlib style for dark theme compatibility
21
  plt.style.use('dark_background')
22
 
 
17
  scaler_y = joblib.load("scaler_y.pkl")
18
  lstm_model = load_model("lstm_revenue_model.keras")
19
 
20
+ expected_features = xgb_clf.get_booster().feature_names
21
+
22
  # Set matplotlib style for dark theme compatibility
23
  plt.style.use('dark_background')
24