YEHTUT commited on
Commit
6fff34e
·
verified ·
1 Parent(s): cdbbad3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -126,7 +126,7 @@ if st.button("Predict"):
126
 
127
  # Load the selected model
128
  if selected_model == 'Logistic Regression':
129
- with open('logistic_regression_model.pkl', 'rb') as f:
130
  model = pickle.load(f)
131
  elif selected_model == 'Decision Tree':
132
  with open('decision_tree_model.pkl', 'rb') as f:
 
126
 
127
  # Load the selected model
128
  if selected_model == 'Logistic Regression':
129
+ with open('best_logreg_model.pkl', 'rb') as f:
130
  model = pickle.load(f)
131
  elif selected_model == 'Decision Tree':
132
  with open('decision_tree_model.pkl', 'rb') as f: