model.py

import joblib
import numpy as np

model = joblib.load("model.pkl")

def predict(input_data):
    features = [
        input_data["Pregnancies"],
        input_data["Glucose"],
        input_data["BloodPressure"],
        input_data["SkinThickness"],
        input_data["Insulin"],
        input_data["BMI"],
        input_data["DiabetesPedigreeFunction"],
        input_data["Age"]
    ]
    X = np.array(features).reshape(1, -1)
    prediction = model.predict(X)[0]
    return {"prediction": int(prediction)}