Upload model.py with huggingface_hub

model.py

@@ -0,0 +1,20 @@
+
+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)}