Upload handler.py

handler.py

@@ -0,0 +1,22 @@
+import joblib
+import pandas as pd
+
+class EndpointHandler:
+    def __init__(self, path=""):
+        # load your model artifact
+        self.artifact = joblib.load(f"{path}/csic_rf_model.joblib")
+
+    def __call__(self, data):
+        # input must be a list in correct feature order
+        inputs = data["inputs"]
+
+        df = pd.DataFrame([inputs], columns=self.artifact["features"])
+
+        pred = self.artifact["model"].predict(df)
+
+        # if target encoder exists
+        if "target_encoder" in self.artifact:
+            label = self.artifact["target_encoder"].inverse_transform(pred)
+            return {"prediction": label[0]}
+
+        return {"prediction": int(pred[0])}