Update main.py

main.py

@@ -6,24 +6,18 @@ import numpy as np
 import os
 
 app = FastAPI()
-#app.mount("/static", StaticFiles(directory="static"), name="static")
-#templates = Jinja2Templates(directory="templates")
 
 def load_model():
-    cwd = os.getcwd()
-    destination = os.path.join(cwd, "Assets")
+    num_imputer_filepath = "numerical_imputer.joblib"
+    scaler_filepath = "scaler.joblib"
+    model_filepath = "lr_model.joblib"
 
-    imputer_filepath = os.path.join(destination, "numerical_imputer.joblib")
-    scaler_filepath = os.path.join(destination, "scaler.joblib")
-    model_filepath = os.path.join(destination, "lr_model.joblib")
-
-    num_imputer = joblib.load(imputer_filepath)
+    num_imputer = joblib.load(num_imputer_filepath)
     scaler = joblib.load(scaler_filepath)
     model = joblib.load(model_filepath)
 
     return num_imputer, scaler, model
 
-
 def preprocess_input_data(input_data, num_imputer, scaler):
     input_data_df = pd.DataFrame([input_data])
     num_columns = [col for col in input_data_df.columns if input_data_df[col].dtype != 'object']
@@ -33,8 +27,20 @@ def preprocess_input_data(input_data, num_imputer, scaler):
 
 @app.get("/")
 def read_root():
+    """
+    Welcome to the Sepsis Prediction API! 🩺💉
+
+    This API provides advanced machine learning predictions for sepsis. ⚡📊
+
+    For more information and to explore the API's capabilities,
+    please visit the documentation: [API Documentation](https://abubakari-sepsis-fastapi-prediction-app.hf.space/docs/) 📚🔍
+
+    Stay proactive in detecting sepsis with our predictive tool! 🚀🔮
+
+    """
     return "Sepsis Prediction App"
 
+
 @app.get("/sepsis/predict")
 def predict_sepsis_endpoint(PRG: float, PL: float, PR: float, SK: float, TS: float,
                             M11: float, BD2: float, Age: float, Insurance: int):