Update app.py
Browse filesSlight changes in the low priority error
app.py
CHANGED
|
@@ -51,15 +51,15 @@ if uploaded_file is not None:
|
|
| 51 |
|
| 52 |
# Load the selected model
|
| 53 |
if model_name == "K-Nearest Neighbors - (Single Label)":
|
| 54 |
-
model = joblib.load("
|
| 55 |
elif model_name == "Logistic Regression - (Single Label)":
|
| 56 |
-
model = joblib.load("
|
| 57 |
elif model_name == "Support Vector Machines - (Single Label)":
|
| 58 |
-
model = joblib.load("
|
| 59 |
elif model_name == "Neural Network - (Single Label)":
|
| 60 |
-
model = joblib.load("
|
| 61 |
elif model_name == "XGB Classifier - (Single Label)":
|
| 62 |
-
model = joblib.load("
|
| 63 |
elif model_name == "XGB - (Multi Label)":
|
| 64 |
model = joblib.load("../models/xgb_mlb.pkl")
|
| 65 |
elif model_name == "Convolutional Recurrent Neural Network - (Multi Label)":
|
|
|
|
| 51 |
|
| 52 |
# Load the selected model
|
| 53 |
if model_name == "K-Nearest Neighbors - (Single Label)":
|
| 54 |
+
model = joblib.load("./models/knn.pkl")
|
| 55 |
elif model_name == "Logistic Regression - (Single Label)":
|
| 56 |
+
model = joblib.load("./models/logistic.pkl")
|
| 57 |
elif model_name == "Support Vector Machines - (Single Label)":
|
| 58 |
+
model = joblib.load("./models/svm.pkl")
|
| 59 |
elif model_name == "Neural Network - (Single Label)":
|
| 60 |
+
model = joblib.load("./models/nn.pkl")
|
| 61 |
elif model_name == "XGB Classifier - (Single Label)":
|
| 62 |
+
model = joblib.load("./models/xgb.pkl")
|
| 63 |
elif model_name == "XGB - (Multi Label)":
|
| 64 |
model = joblib.load("../models/xgb_mlb.pkl")
|
| 65 |
elif model_name == "Convolutional Recurrent Neural Network - (Multi Label)":
|