| from flask import Flask, request, jsonify |
| import numpy as np |
| import xgboost as xgb |
|
|
| app = Flask(__name__) |
|
|
| |
| model = xgb.Booster() |
| model.load_model("xgb_superkart_model.json") |
|
|
| @app.route("/", methods=["GET"]) |
| def root(): |
| return "SuperKart Sales Forecasting API is running." |
|
|
|
|
| @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) |
|
|
