Update main.py

main.py

@@ -10,8 +10,7 @@ from pydantic import BaseModel  # Import Pydantic's BaseModel
 app = FastAPI()
 templates = Jinja2Templates(directory="templates")
 
-# Load the pickled XGBoost model
-xgb_model = joblib.load("model.joblib")
+
 
 class InputFeatures(BaseModel):
     prg: float
@@ -23,15 +22,19 @@ class InputFeatures(BaseModel):
     bd2: float
     age: int
 
+# Load the pickled XGBoost model
+xgb_model = joblib.load("model.joblib")
+
+
 @app.get("/")
 async def read_root():
-    return {"message": "Welcome to the XGBoost Diabetes Prediction API"}
+    return {"message": "Welcome to the Sepsis Prediction API"}
 
 @app.get("/form/")
 async def show_form():
 
 @app.post("/predict/")
-async def predict_diabetes(
+async def predict_sepsis(
     request: Request,
     input_data: InputFeatures  # Use the Pydantic model for input validation
 ):