Spaces:
Sleeping
Sleeping
upload files
Browse files- .gitignore +1 -0
- Dockerfile +14 -0
- app/__pycache__/main.cpython-311.pyc +0 -0
- app/main.py +50 -0
- models/sepssis_model_v1.pkl +3 -0
- requirements.txt +5 -0
.gitignore
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
myvenv
|
Dockerfile
ADDED
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
FROM python:3.9-slim
|
2 |
+
|
3 |
+
WORKDIR /app
|
4 |
+
|
5 |
+
COPY ./models/sepssis_model_v1.pkl /app/models/sepssis_model_v1.pkl
|
6 |
+
Copy ./requirements.txt /app/requirements.txt
|
7 |
+
|
8 |
+
RUN pip install --no-cache-dir -r requirements.txt
|
9 |
+
|
10 |
+
COPY ./app /app
|
11 |
+
|
12 |
+
EXPOSE 8000
|
13 |
+
|
14 |
+
CMD ["uvicorn", "main:app", "--host","0.0.0.0","--port","8000"]
|
app/__pycache__/main.cpython-311.pyc
ADDED
Binary file (1.81 kB). View file
|
|
app/main.py
ADDED
@@ -0,0 +1,50 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from fastapi import FastAPI
|
2 |
+
from pydantic import BaseModel
|
3 |
+
import joblib
|
4 |
+
import numpy as np
|
5 |
+
|
6 |
+
|
7 |
+
class SepssisInput(BaseModel):
|
8 |
+
PRG:float
|
9 |
+
PL:float
|
10 |
+
PR:float
|
11 |
+
SK:float
|
12 |
+
TS:float
|
13 |
+
M11:float
|
14 |
+
Age:float
|
15 |
+
Insurance:int
|
16 |
+
|
17 |
+
|
18 |
+
model=joblib.load("models/sepssis_model_v1.pkl")
|
19 |
+
|
20 |
+
app=FastAPI()
|
21 |
+
|
22 |
+
|
23 |
+
@app.get("/")
|
24 |
+
def home_page():
|
25 |
+
return "Welcome to the API "
|
26 |
+
|
27 |
+
|
28 |
+
@app.post("/predict")
|
29 |
+
def inference_sepssis(sepssis_features:SepssisInput):
|
30 |
+
input_data=np.array([[
|
31 |
+
sepssis_features.PRG,
|
32 |
+
sepssis_features.PL,
|
33 |
+
sepssis_features.PR,
|
34 |
+
sepssis_features.SK,
|
35 |
+
sepssis_features.TS,
|
36 |
+
sepssis_features.M11,
|
37 |
+
sepssis_features.Age,
|
38 |
+
sepssis_features.Insurance
|
39 |
+
]])
|
40 |
+
|
41 |
+
label_predict=model.predict(input_data)
|
42 |
+
|
43 |
+
sepssis_mapping={
|
44 |
+
0:"Negative",
|
45 |
+
1:"Positive"
|
46 |
+
}
|
47 |
+
|
48 |
+
clase_predict=sepssis_mapping[label_predict[0]]
|
49 |
+
return {"Predict_for_Sepssis":clase_predict}
|
50 |
+
|
models/sepssis_model_v1.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4cd86d3763f4106af99664f1b62b7cde8eb857de037c0dbfde9f1bc142d511a7
|
3 |
+
size 856265
|
requirements.txt
ADDED
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
joblib==1.3.2
|
2 |
+
scikit-learn==1.2.2
|
3 |
+
fastapi
|
4 |
+
uvicorn
|
5 |
+
|