Upload folder using huggingface_hub

Dockerfile

@@ -0,0 +1,20 @@
+# Base image with Python
+FROM python:3.10-slim
+
+# Set working directory inside container
+WORKDIR /app
+
+# Copy requirements first (for caching)
+COPY requirements.txt .
+
+# Install dependencies
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy all project files (API + model)
+COPY . .
+
+# Expose port for Flask
+EXPOSE 7860
+
+# Run the Flask app
+CMD ["python", "app.py"]

app.py

@@ -0,0 +1,42 @@
+
+from flask import Flask, request, jsonify
+import joblib
+import numpy as np
+
+
+from huggingface_hub import hf_hub_download
+import joblib
+
+# download from model repo
+model_path = hf_hub_download(
+    repo_id="affanthinks/superkart",
+    filename="tuned_bagging_model.pkl"
+)
+
+model = joblib.load(model_path)
+
+
+# Initialize app
+app = Flask("predict_revenue")
+
+@app.route("/")
+def home():
+    return jsonify({"message": "Supermarket Revenue Prediction API is running!"})
+
+@app.route("/predict", methods=["POST"])
+def predict():
+    try:
+        # Get JSON input
+        data = request.get_json(force=True)
+        features = np.array(data["features"]).reshape(1, -1)
+
+        # Predict
+        prediction = model.predict(features)[0]
+
+        return jsonify({"predicted_revenue": float(prediction)})
+
+    except Exception as e:
+        return jsonify({"error": str(e)})
+
+if "predict_revenue" == "__main__":
+    app.run(host="0.0.0.0", port=7860, debug=True)

requirements.txt

@@ -0,0 +1,5 @@
+flask
+numpy==1.26.4
+pandas==2.2.2
+scikit-learn==1.5.1
+joblib==1.4.2