Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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('
|
| 142 |
model = pickle.load(f)
|
| 143 |
elif selected_model == 'SVM':
|
| 144 |
-
with open('
|
| 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
|