Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -143,7 +143,7 @@ def train():
|
|
| 143 |
# Save model and vectorizer
|
| 144 |
joblib.dump(model, config.MODEL_PATH)
|
| 145 |
joblib.dump(vectorizer, config.TFIDF_VECTORIZER_PATH)
|
| 146 |
-
|
| 147 |
return {
|
| 148 |
"message": "Training completed successfully.",
|
| 149 |
"accuracy": accuracy
|
|
|
|
| 143 |
# Save model and vectorizer
|
| 144 |
joblib.dump(model, config.MODEL_PATH)
|
| 145 |
joblib.dump(vectorizer, config.TFIDF_VECTORIZER_PATH)
|
| 146 |
+
joblib.dump(label_encoders, config.LABEL_ENCODERS_PATH)
|
| 147 |
return {
|
| 148 |
"message": "Training completed successfully.",
|
| 149 |
"accuracy": accuracy
|