Sonny4Sonnix commited on
Commit
67ca5ef
·
1 Parent(s): 06a1da0

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
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("xgb_model.joblib")
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