push files to HuggingFace Hub

.dockerignore

@@ -0,0 +1 @@
+.venv

dockerfile

@@ -0,0 +1,29 @@
+# Use of an official Python runtime as a base image
+FROM python:3.11.5
+
+# #Set up of working directory in the container i.e /app
+# WORKDIR /code
+
+# # Copy current directory contents into the container at /app
+# COPY . /app
+
+# # Install needed packages specified in requirements.txt recursively
+# RUN pip install -r requirements.txt
+
+# # Expose the port number the app runs on
+# EXPOSE 7860
+
+# # Command to run the application
+#CMD ["uvicorn", "src.main:app", "--host", "0.0.0.0", "--port", "7860"]
+
+FROM python:3.11.5
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY . .
+
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -0,0 +1,98 @@
+from fastapi import FastAPI, HTTPException
+import uvicorn
+#from typing import List, Literal
+from pydantic import BaseModel, Field
+import pandas as pd
+import pickle, os
+
+ #setup
+# Get the directory of the current file (FastAPI application file)
+DIRPATH = os.path.dirname(os.path.realpath(__file__))
+
+# Construct the path to ml.pkl relative to the current file using forward slashes
+ml_core_fp = os.path.join(DIRPATH, "../model/ml.pkl")
+
+
+#useful functions
+def  load_ml_components(fp):
+    "load the ml components to re-use in app"
+    with open(fp, 'rb') as file:
+        obj = pickle.load(file)
+        return obj
+    
+
+# Loading: Execute and instantiate ml components
+ml_components_dict = load_ml_components(fp = ml_core_fp)
+
+pipeline = ml_components_dict["pipeline"]
+
+encoder =  ml_components_dict["encoder"]
+
+
+# API
+app = FastAPI(
+    title= "Sepsis classification API"
+)
+
+# Input for Modelling 
+class Sepsis_Pred(BaseModel):
+
+    PRG: int = Field(..., description='Plasma glucose')
+    PL: int = Field(..., description='Blood Work Result-1 (mu U/ml)')
+    PR: int = Field(..., description='Blood Pressure (mm Hg)')
+    SK: int = Field(..., description='Blood Work Result-2 (mm)')
+    TS: int = Field(..., description='Blood Work Result-3 (mu U/ml)')
+    M11: float = Field(..., description='Body mass index (weight in kg/(height in m)^2)')
+    BD2: float = Field(..., description='Blood Work Result-4 (mu U/ml)')
+    Age: int = Field(..., description='Patient age (years)')
+    Insurance: int = Field(..., description='If a patient holds a valid insurance card')
+
+
+@app.get("/")
+def root():
+    return {
+        "Info": "Sepsis classification API : This API classifies whether a patient will develop sepsis based on various test results"
+    }
+
+
+@app.post("/classify_patient")
+def sepsis_classification(sepsis_pred: Sepsis_Pred):
+
+    try:
+
+        #Dataframe creation
+        df = pd.DataFrame([sepsis_pred.model_dump()])
+
+        print(f'df: {df}')
+       
+        # ML prediction
+        prediction = pipeline.predict(df)
+
+        # Get the index of the predicted class (0 or 1 in binary classification)
+        predicted_class_index = prediction[0]
+
+        confidence_score = pipeline.predict_proba(df)
+
+        # Retrieve the confidence score for the predicted class
+
+        confidence_score_predicted_class = confidence_score[0][predicted_class_index]
+
+
+        print(f"confidence_score: {confidence_score}")
+
+        execution_message = "Execution successful"
+
+
+        # encoded prediction
+        decoded_prediction = encoder.inverse_transform([prediction])[0]
+
+    
+        return {"execution message": execution_message, "patient_diagnosis": decoded_prediction, "confidence_score": confidence_score_predicted_class}
+    
+    
+    except Exception as e: 
+        raise HTTPException(status_code=500, detail=f"An error occurred during prediction {str(e)}")
+
+
+if __name__ == "__main__":
+    uvicorn.run("main:app", reload=True)

requirements.txt

@@ -0,0 +1,51 @@
+annotated-types==0.6.0
+anyio==3.7.1
+certifi==2023.11.17
+click==8.1.7
+colorama==0.4.6
+contourpy==1.2.0
+cycler==0.12.1
+dnspython==2.4.2
+email-validator==2.1.0.post1
+fastapi==0.105.0
+fonttools==4.46.0
+h11==0.14.0
+httpcore==1.0.2
+httptools==0.6.1
+httpx==0.25.2
+idna==3.6
+imbalanced-learn==0.11.0
+itsdangerous==2.1.2
+Jinja2==3.1.2
+joblib==1.3.2
+kiwisolver==1.4.5
+MarkupSafe==2.1.3
+matplotlib==3.8.2
+numpy==1.26.2
+orjson==3.9.10
+packaging==23.2
+pandas==2.1.4
+Pillow==10.1.0
+pydantic==2.5.2
+pydantic-extra-types==2.2.0
+pydantic-settings==2.1.0
+pydantic_core==2.14.5
+pyparsing==3.1.1
+python-dateutil==2.8.2
+python-dotenv==1.0.0
+python-multipart==0.0.6
+pytz==2023.3.post1
+PyYAML==6.0.1
+scikit-learn==1.3.2
+scipy==1.11.4
+seaborn==0.13.0
+six==1.16.0
+sniffio==1.3.0
+starlette==0.27.0
+threadpoolctl==3.2.0
+typing_extensions==4.9.0
+tzdata==2023.3
+ujson==5.9.0
+uvicorn==0.24.0.post1
+watchfiles==0.21.0
+websockets==12.0