Update app.py

app.py

@@ -1,19 +1,20 @@
-from flask import Flask, request, jsonify
-import pickle
-import numpy as np
-
-app = Flask(__name__)
-
-# Load model
-with open("xgb_superkart_model.pkl", "rb") as f:
-    model = pickle.load(f)
-
-@app.route("/predict", methods=["POST"])
-def predict():
-    data = request.get_json(force=True)
-    features = np.array(data["features"]).reshape(1, -1)
-    prediction = model.predict(features)
-    return jsonify({"prediction": float(prediction[0])})
-
-if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
+from flask import Flask, request, jsonify
+import numpy as np
+import xgboost as xgb
+
+app = Flask(__name__)
+
+# Load model from JSON (instead of pickle)
+model = xgb.Booster()
+model.load_model("model/xgb_superkart_model.json")
+
+@app.route("/predict", methods=["POST"])
+def predict():
+    data = request.get_json(force=True)
+    features = np.array(data["features"]).reshape(1, -1)
+    dmatrix = xgb.DMatrix(features)
+    prediction = model.predict(dmatrix)
+    return jsonify({"prediction": float(prediction[0])})
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)