Spaces:
Runtime error
Runtime error
fix(prediction): correct model name for l2
Browse files
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/
|
| 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"]
|