File size: 853 Bytes
c29a8e4 c915b5c c29a8e4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
from flask import Flask, request, jsonify
from flask_cors import CORS
import pandas as pd, joblib, os
app = Flask(__name__)
CORS(app)
model = joblib.load("final_random_forest_model.pkl")
FEATURE_COLUMNS = [
"Product_Weight", "Product_Allocated_Area", "Product_MRP",
"Store_Establishment_Year", "Store_Size", "Store_Location_City_Type",
"Store_Type", "Product_Prefix", "Product_Num", "Store_Age"
]
@app.route("/", methods=["GET"])
def home():
return "✅ SuperKart Forecast API is running"
@app.route("/predict", methods=["POST"])
def predict():
data = request.get_json(force=True)["data"]
df = pd.DataFrame(data)[FEATURE_COLUMNS]
preds = model.predict(df)
return jsonify({"predictions": preds.tolist()})
if __name__ == "__main__":
port = int(os.environ.get("PORT", 7860))
app.run(host="0.0.0.0", port=port)
|