subbunanepalli commited on
Commit
fc94fea
·
verified ·
1 Parent(s): 0aed451

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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