Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 =
|
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]
|