subbunanepalli commited on
Commit
394ba43
·
verified ·
1 Parent(s): 4f6579b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -211,7 +211,7 @@ async def validate_model(
211
  model_path = MODEL_SAVE_DIR / f"{model_name}.pkl"
212
  if not model_path.exists():
213
  raise HTTPException(status_code=404, detail="XGB model file not found")
214
- model = TfidfXGB(label_encoders)
215
  model.load_model(model_name)
216
  X = data_df[TEXT_COLUMN]
217
  y = data_df[LABEL_COLUMNS]
 
211
  model_path = MODEL_SAVE_DIR / f"{model_name}.pkl"
212
  if not model_path.exists():
213
  raise HTTPException(status_code=404, detail="XGB model file not found")
214
+ model = TfidfXGBoost(label_encoders)
215
  model.load_model(model_name)
216
  X = data_df[TEXT_COLUMN]
217
  y = data_df[LABEL_COLUMNS]