| from flask import Flask, request, jsonify
|
| import pickle
|
| import numpy as np
|
|
|
| app = Flask(__name__)
|
|
|
|
|
| with open("xgb_superkart_model.pkl", "rb") as f:
|
| model = pickle.load(f)
|
|
|
| @app.route("/")
|
| def home():
|
| return "SuperKart Sales Forecast API is Live"
|
|
|
| @app.route("/predict", methods=["POST"])
|
| def predict():
|
| data = request.get_json(force=True)
|
| input_features = np.array(data["features"]).reshape(1, -1)
|
| prediction = model.predict(input_features)
|
| return jsonify({"predicted_sales": prediction[0]})
|
|
|
| if __name__ == "__main__":
|
| app.run(host='0.0.0.0', port=7860)
|
|
|
