Upload folder using huggingface_hub

Dockerfile

@@ -0,0 +1,6 @@
+FROM python:3.9-slim
+WORKDIR /app
+COPY . .
+RUN pip install -r requirements.txt
+EXPOSE 8000
+CMD ["python", "app.py"]

app.py

@@ -0,0 +1,24 @@
+from flask import Flask, request, jsonify
+import joblib
+import numpy as np
+import pandas as pd
+
+app = Flask(__name__)
+model = joblib.load("superkart_best_model.pkl")
+
+FEATURE_COLUMNS = ["Product_Weight","Product_Sugar_Content","Product_Allocated_Area","Product_Type","Product_MRP","Store_Establishment_Year","Store_Size","Store_Location_City_Type","Store_Type","Store_Current_Age","Product_Type_Group"]
+
+@app.route("/predict", methods=["POST"])
+def predict():
+    data = request.get_json()
+    df_input = pd.DataFrame([data])
+    df_input = df_input.reindex(columns=FEATURE_COLUMNS, fill_value=0)
+    preds = model.predict(df_input)
+    return jsonify({"predictions": preds.tolist()})
+
+@app.route("/", methods=["GET"])
+def home():
+    return "SuperKart API Ready!"
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=8000)

requirements.txt

@@ -0,0 +1,6 @@
+flask==3.0.3
+numpy==2.0.2
+pandas==2.2.2
+scikit-learn==1.6.1
+xgboost==2.1.4
+joblib==1.4.2

superkart_best_model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d09945964f4a7e3d3983192be044bb459d5540d92ad9a0b979c8e4e5c2b256c5
+size 1270901