YEHTUT commited on
Commit
9449905
·
verified ·
1 Parent(s): 2acf314

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -138,10 +138,10 @@ if st.button("Predict"):
138
  with open('xgboost_model.pkl', 'rb') as f:
139
  model = pickle.load(f)
140
  elif selected_model == 'Random Forest':
141
- with open('rf_clf.pkl', 'rb') as f:
142
  model = pickle.load(f)
143
  elif selected_model == 'SVM':
144
- with open('svm_model.pkl', 'rb') as f:
145
  model = pickle.load(f)
146
 
147
  # Make predictions
 
138
  with open('xgboost_model.pkl', 'rb') as f:
139
  model = pickle.load(f)
140
  elif selected_model == 'Random Forest':
141
+ with open('best_rf_model.pkl', 'rb') as f:
142
  model = pickle.load(f)
143
  elif selected_model == 'SVM':
144
+ with open('best_svm_model.pkl', 'rb') as f:
145
  model = pickle.load(f)
146
 
147
  # Make predictions