Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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('
|
| 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:
|