testing-models / projects /ML_DiabetesPrediction /ML_DiabetesPrediction.py
eziokittu's picture
Update projects/ML_DiabetesPrediction/ML_DiabetesPrediction.py
c043326 verified
raw
history blame contribute delete
380 Bytes
from pydantic import BaseModel
import pickle
# loading the saved model
model_ML_DiabetesPrediction = pickle.load(open('./projects/ML_DiabetesPrediction/model_DiabetesPrediction.pkl', 'rb'))
class form2(BaseModel):
Pregnancies: int
Glucose: int
BloodPressure: int
SkinThickness: int
Insulin: int
BMI: float
DiabetesPedigreeFunction: float
Age: int