Thanut003 commited on
Commit
fd74610
·
verified ·
1 Parent(s): 0e17d47

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -225,31 +225,31 @@ RED_SVD = "preprocessor/truncated_svd.joblib"
225
 
226
  # Map each model to its specific file paths
227
  MODEL_CONFIG = {
228
- "XGBoost": {
229
  "model_path": "models/bow_models_without_pca/xgboost_model.joblib",
230
  "vec_path": VEC_COUNT,
231
  "red_path": None,
232
  "dense_required": False
233
  },
234
- "LightGBM": {
235
  "model_path": "models/bow_models_without_pca/lightgbm_model.joblib",
236
  "vec_path": VEC_COUNT,
237
  "red_path": None,
238
  "dense_required": False
239
  },
240
- "Random Forest": {
241
  "model_path": "models/bow_models_without_pca/random_forest_model.joblib",
242
  "vec_path": VEC_COUNT,
243
  "red_path": None,
244
  "dense_required": False
245
  },
246
- "Linear SVM": {
247
  "model_path": "models/tfidf_models_with_truncatedSVD/linear_svm_model.joblib",
248
  "vec_path": VEC_TFIDF,
249
  "red_path": RED_SVD,
250
  "dense_required": False
251
  },
252
- "Logistic Regression": {
253
  "model_path": "models/tfidf_models_with_truncatedSVD/logistic_regression_model.joblib",
254
  "vec_path": VEC_TFIDF,
255
  "red_path": RED_SVD,
 
225
 
226
  # Map each model to its specific file paths
227
  MODEL_CONFIG = {
228
+ "XGBoost (BoW)": {
229
  "model_path": "models/bow_models_without_pca/xgboost_model.joblib",
230
  "vec_path": VEC_COUNT,
231
  "red_path": None,
232
  "dense_required": False
233
  },
234
+ "LightGBM (BoW)": {
235
  "model_path": "models/bow_models_without_pca/lightgbm_model.joblib",
236
  "vec_path": VEC_COUNT,
237
  "red_path": None,
238
  "dense_required": False
239
  },
240
+ "Random Forest (BoW)": {
241
  "model_path": "models/bow_models_without_pca/random_forest_model.joblib",
242
  "vec_path": VEC_COUNT,
243
  "red_path": None,
244
  "dense_required": False
245
  },
246
+ "Linear SVM (TF-IDF + SVD)": {
247
  "model_path": "models/tfidf_models_with_truncatedSVD/linear_svm_model.joblib",
248
  "vec_path": VEC_TFIDF,
249
  "red_path": RED_SVD,
250
  "dense_required": False
251
  },
252
+ "Logistic Regression (TF-IDF + SVD)": {
253
  "model_path": "models/tfidf_models_with_truncatedSVD/logistic_regression_model.joblib",
254
  "vec_path": VEC_TFIDF,
255
  "red_path": RED_SVD,