Commit
·
67ca5ef
1
Parent(s):
06a1da0
Update main.py
Browse files
main.py
CHANGED
@@ -11,7 +11,7 @@ app = FastAPI()
|
|
11 |
templates = Jinja2Templates(directory="templates")
|
12 |
|
13 |
# Load the pickled XGBoost model
|
14 |
-
xgb_model = joblib.load("
|
15 |
|
16 |
class InputFeatures(BaseModel):
|
17 |
prg: float
|
|
|
11 |
templates = Jinja2Templates(directory="templates")
|
12 |
|
13 |
# Load the pickled XGBoost model
|
14 |
+
xgb_model = joblib.load("model.joblib")
|
15 |
|
16 |
class InputFeatures(BaseModel):
|
17 |
prg: float
|