import pandas as pd
import pickle5 as pickle
from fastapi import FastAPI
from pydantic import BaseModel
import numpy as np
app=FastAPI()
class diabetes(BaseModel):
Pregnancies:float
Glucose:float
BloodPressure:float
SkinThickness:float
Insulin:float
BMI:float
DiabetesPedigreeFunction:float
Age:float
with open("diabetes.sav",'rb') as f:
model = pickle.load(f)
@app.post("/")
async def diabetes(data:diabetes):
test_data = np.array([
data.Pregnancies,
data.Glucose,
data.BloodPressure,
data.SkinThickness,
data.Insulin,
data.BMI,
data.DiabetesPedigreeFunction,
data.Age
]).reshape(1,-1)
pred =model.predict(test_data)
return {"prediction":pred}