Mekam commited on
Commit
64ec546
·
1 Parent(s): 724b321

fix(prediction): correct model name for l2

Browse files
Files changed (1) hide show
  1. src/agents/l2_supervisor.py +1 -1
src/agents/l2_supervisor.py CHANGED
@@ -2,7 +2,7 @@ import joblib
2
  from fastapi import HTTPException
3
 
4
  class Supervisor:
5
- def __init__(self, model_path: str = "src/models/L2_XGBOOST_Supervisor_V0.joblib"):
6
  try:
7
  saved = joblib.load(model_path)
8
  self.model = saved["model"]
 
2
  from fastapi import HTTPException
3
 
4
  class Supervisor:
5
+ def __init__(self, model_path: str = "src/models/l2_XGBOOST_Supervisor_V0.joblib"):
6
  try:
7
  saved = joblib.load(model_path)
8
  self.model = saved["model"]